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 build error on cuda10.2 platform (#1597)

Browse Source
This commit is contained in:
Chen Xin 2022-12-30 19:48:55 +08:00 committed by GitHub
parent 09f1952457
commit c67e2db68e
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 <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,
@ -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