fix cuda10.2 build (#1539)

pull/1554/head
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

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