Skip to content

Commit

Permalink
Fix build error on cuda10.2 platform (#1597)
Browse files Browse the repository at this point in the history
  • Loading branch information
irexyc authored Dec 30, 2022
1 parent 09f1952 commit c67e2db
Showing 1 changed file with 6 additions and 2 deletions.
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

0 comments on commit c67e2db

Please sign in to comment.