未验证 提交 82fb63eb 编写于 作者: W wangxinxin08 提交者: GitHub

fix dcnv2 trt8 compile error (#36850)

上级 f3ee5c99
...@@ -360,7 +360,7 @@ void gemm_impl<half>(cublasHandle_t handle, cublasOperation_t transa, ...@@ -360,7 +360,7 @@ void gemm_impl<half>(cublasHandle_t handle, cublasOperation_t transa,
template <typename T> template <typename T>
int DeformableConvPlugin::enqueue_impl(int batch_size, int DeformableConvPlugin::enqueue_impl(int batch_size,
const void* const* inputs, const void* const* inputs,
void** outputs, void* workspace, void* const* outputs, void* workspace,
cudaStream_t stream) { cudaStream_t stream) {
const T* input = reinterpret_cast<const T*>(inputs[0]); const T* input = reinterpret_cast<const T*>(inputs[0]);
const T* offset = reinterpret_cast<const T*>(inputs[1]); const T* offset = reinterpret_cast<const T*>(inputs[1]);
...@@ -527,8 +527,6 @@ nvinfer1::IPluginV2Ext* DeformableConvPlugin::clone() const TRT_NOEXCEPT { ...@@ -527,8 +527,6 @@ nvinfer1::IPluginV2Ext* DeformableConvPlugin::clone() const TRT_NOEXCEPT {
offset_dim_, mask_dim_, output_dim_); offset_dim_, mask_dim_, output_dim_);
} }
DeformableConvPluginCreator::DeformableConvPluginCreator() TRT_NOEXCEPT {}
void DeformableConvPluginCreator::setPluginNamespace(const char* lib_namespace) void DeformableConvPluginCreator::setPluginNamespace(const char* lib_namespace)
TRT_NOEXCEPT { TRT_NOEXCEPT {
namespace_ = std::string(lib_namespace); namespace_ = std::string(lib_namespace);
......
...@@ -91,8 +91,8 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext { ...@@ -91,8 +91,8 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext {
private: private:
template <typename T> template <typename T>
int enqueue_impl(int batch_size, const void* const* inputs, void** outputs, int enqueue_impl(int batch_size, const void* const* inputs,
void* workspace, cudaStream_t stream); void* const* outputs, void* workspace, cudaStream_t stream);
nvinfer1::Weights copyToDevice(const void* hostData, size_t count); nvinfer1::Weights copyToDevice(const void* hostData, size_t count);
void serializeFromDevice(void** hostBuffer, void serializeFromDevice(void** hostBuffer,
const nvinfer1::Weights& deviceWeights) const; const nvinfer1::Weights& deviceWeights) const;
...@@ -119,7 +119,7 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext { ...@@ -119,7 +119,7 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext {
class DeformableConvPluginCreator : public nvinfer1::IPluginCreator { class DeformableConvPluginCreator : public nvinfer1::IPluginCreator {
public: public:
DeformableConvPluginCreator(); DeformableConvPluginCreator() = default;
~DeformableConvPluginCreator() override = default; ~DeformableConvPluginCreator() override = default;
void setPluginNamespace(const char* lib_namespace) TRT_NOEXCEPT override; void setPluginNamespace(const char* lib_namespace) TRT_NOEXCEPT override;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册