未验证 提交 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,
template <typename T>
int DeformableConvPlugin::enqueue_impl(int batch_size,
const void* const* inputs,
void** outputs, void* workspace,
void* const* outputs, void* workspace,
cudaStream_t stream) {
const T* input = reinterpret_cast<const T*>(inputs[0]);
const T* offset = reinterpret_cast<const T*>(inputs[1]);
......@@ -527,8 +527,6 @@ nvinfer1::IPluginV2Ext* DeformableConvPlugin::clone() const TRT_NOEXCEPT {
offset_dim_, mask_dim_, output_dim_);
}
DeformableConvPluginCreator::DeformableConvPluginCreator() TRT_NOEXCEPT {}
void DeformableConvPluginCreator::setPluginNamespace(const char* lib_namespace)
TRT_NOEXCEPT {
namespace_ = std::string(lib_namespace);
......
......@@ -91,8 +91,8 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext {
private:
template <typename T>
int enqueue_impl(int batch_size, const void* const* inputs, void** outputs,
void* workspace, cudaStream_t stream);
int enqueue_impl(int batch_size, const void* const* inputs,
void* const* outputs, void* workspace, cudaStream_t stream);
nvinfer1::Weights copyToDevice(const void* hostData, size_t count);
void serializeFromDevice(void** hostBuffer,
const nvinfer1::Weights& deviceWeights) const;
......@@ -119,7 +119,7 @@ class DeformableConvPlugin : public nvinfer1::IPluginV2Ext {
class DeformableConvPluginCreator : public nvinfer1::IPluginCreator {
public:
DeformableConvPluginCreator();
DeformableConvPluginCreator() = default;
~DeformableConvPluginCreator() override = default;
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.
先完成此消息的编辑!
想要评论请 注册