mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

fix cuda10.2 build (#1539)

Browse Source
This commit is contained in:
Chen Xin 2022-12-16 10:15:15 +08:00 committed by GitHub
parent 77c2ee5feb
commit dbc4b26dc1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
csrc/mmdeploy/codebase/mmaction/cuda/transpose.cu
View File

@ -3,7 +3,9 @@
#include <stdint.h> #include <stdint.h>
#include <stdio.h> #include <stdio.h>
namespace mmdeploy::mmaction::cuda { namespace mmdeploy {
namespace mmaction {
namespace cuda {
template <typename T> template <typename T>
__global__ void transpose(const T* src, const int* src_strides, T* dst, const int* dst_strides, __global__ void transpose(const T* src, const int* src_strides, T* dst, const int* dst_strides,
@ -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, template void Transpose<float>(const float* src, const int* src_strides, float* dst,
const int* dst_strides, int ndim, int total, cudaStream_t stream); const int* dst_strides, int ndim, int total, cudaStream_t stream);
} // namespace mmdeploy::mmaction::cuda } // namespace cuda
} // namespace mmaction
} // namespace mmdeploy