Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions csrc/mmdeploy/codebase/mmaction/cuda/transpose.cu
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,9 @@
#include <stdint.h>
#include <stdio.h>

namespace mmdeploy::mmaction::cuda {
namespace mmdeploy {
namespace mmaction {
namespace cuda {

template <typename T>
__global__ void transpose(const T* src, const int* src_strides, T* dst, const int* dst_strides,
Expand Down Expand Up @@ -35,4 +37,6 @@ void Transpose(const T* src, const int* src_strides, T* dst, const int* dst_stri
template void Transpose<float>(const float* src, const int* src_strides, float* dst,
const int* dst_strides, int ndim, int total, cudaStream_t stream);

} // namespace mmdeploy::mmaction::cuda
} // namespace cuda
} // namespace mmaction
} // namespace mmdeploy