Skip to content

Commit 82fb63e

Browse files
authored
fix dcnv2 trt8 compile error (#36850)
1 parent f3ee5c9 commit 82fb63e

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

paddle/fluid/inference/tensorrt/plugin/deformable_conv_op_plugin.cu

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ void gemm_impl<half>(cublasHandle_t handle, cublasOperation_t transa,
360360
template <typename T>
361361
int DeformableConvPlugin::enqueue_impl(int batch_size,
362362
const void* const* inputs,
363-
void** outputs, void* workspace,
363+
void* const* outputs, void* workspace,
364364
cudaStream_t stream) {
365365
const T* input = reinterpret_cast<const T*>(inputs[0]);
366366
const T* offset = reinterpret_cast<const T*>(inputs[1]);
@@ -527,8 +527,6 @@ nvinfer1::IPluginV2Ext* DeformableConvPlugin::clone() const TRT_NOEXCEPT {
527527
offset_dim_, mask_dim_, output_dim_);
528528
}
529529

530-
DeformableConvPluginCreator::DeformableConvPluginCreator() TRT_NOEXCEPT {}
531-
532530
void DeformableConvPluginCreator::setPluginNamespace(const char* lib_namespace)
533531
TRT_NOEXCEPT {
534532
namespace_ = std::string(lib_namespace);

paddle/fluid/inference/tensorrt/plugin/deformable_conv_op_plugin.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,8 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext {
9191

9292
private:
9393
template <typename T>
94-
int enqueue_impl(int batch_size, const void* const* inputs, void** outputs,
95-
void* workspace, cudaStream_t stream);
94+
int enqueue_impl(int batch_size, const void* const* inputs,
95+
void* const* outputs, void* workspace, cudaStream_t stream);
9696
nvinfer1::Weights copyToDevice(const void* hostData, size_t count);
9797
void serializeFromDevice(void** hostBuffer,
9898
const nvinfer1::Weights& deviceWeights) const;
@@ -119,7 +119,7 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext {
119119

120120
class DeformableConvPluginCreator : public nvinfer1::IPluginCreator {
121121
public:
122-
DeformableConvPluginCreator();
122+
DeformableConvPluginCreator() = default;
123123
~DeformableConvPluginCreator() override = default;
124124

125125
void setPluginNamespace(const char* lib_namespace) TRT_NOEXCEPT override;

0 commit comments

Comments
 (0)