fix cuda10.2 build (#1539)
parent
77c2ee5feb
commit
dbc4b26dc1
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue