mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
8408ae2bcf
* add multi level roi align * better trt plugin code * fix lint * optimizer trt_multi_level_roi_align.cpp * fix lint
13 lines
490 B
C++
13 lines
490 B
C++
#ifndef TRT_SCATTERND_KERNEL_HPP
|
|
#define TRT_SCATTERND_KERNEL_HPP
|
|
#include <cuda_runtime.h>
|
|
|
|
template <typename T>
|
|
void TRTONNXScatterNDKernelLauncher(const T* data, const int* indices,
|
|
const T* update, const int* dims,
|
|
int nbDims, const int* indices_dims,
|
|
int indice_nbDims, T* output,
|
|
cudaStream_t stream);
|
|
|
|
#endif // TRT_SCATTERND_KERNEL_HPP
|