SeanWangJS / grid-sample3d-trt-plugin

TensorRT plugin for 3-dimension grid sample operator
Apache License 2.0
12 stars 3 forks source link

编译时出错 #1

Open ardyli opened 6 months ago

ardyli commented 6 months ago

(pytorch_env) root@GPUnode01:/data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/build# make [ 20%] Building CXX object CMakeFiles/grid_sample_3d_plugin.dir/src/grid_sample_3d_plugin.cpp.o In file included from /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.cpp:8: /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:17:54: error: expected class-name before ‘{’ token 17 | class GridSample3DPlugin : public IPluginV2DynamicExt{ | ^ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:47:15: error: ‘IPluginV2DynamicExt’ in namespace ‘nvinfer1’ does not name a type; did you mean ‘IPluginV2Ext’? 47 | nvinfer1::IPluginV2DynamicExt clone() const noexcept override; | ^~~~~~~ | IPluginV2Ext /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:49:15: error: ‘DimsExprs’ in namespace ‘nvinfer1’ does not name a type 49 | nvinfer1::DimsExprs getOutputDimensions(int32_t outputIndex, | ^~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:55:36: error: ‘PluginTensorDesc’ has not been declared 55 | PluginTensorDesc const inOut, | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:59:26: error: ‘DynamicPluginTensorDesc’ has not been declared 59 | void configurePlugin(DynamicPluginTensorDesc const in, | ^~~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:61:26: error: ‘DynamicPluginTensorDesc’ has not been declared 61 | DynamicPluginTensorDesc const out, | ^~~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:64:29: error: ‘PluginTensorDesc’ has not been declared 64 | size_t getWorkspaceSize(PluginTensorDesc const inputs, | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:66:29: error: ‘PluginTensorDesc’ has not been declared 66 | PluginTensorDesc const outputs, | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:69:21: error: ‘PluginTensorDesc’ has not been declared 69 | int32_t enqueue(PluginTensorDesc const inputDesc, | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:70:21: error: ‘PluginTensorDesc’ has not been declared 70 | PluginTensorDesc const outputDesc, | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:44:5: error: ‘nvinfer1::plugin::GridSample3DPlugin::~GridSample3DPlugin()’ marked ‘override’, but does not override 44 | ~GridSample3DPlugin() override; | ^ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:54:10: error: ‘bool nvinfer1::plugin::GridSample3DPlugin::supportsFormatCombination(int32_t, const int, int32_t, int32_t)’ marked ‘override’, but does not override 54 | bool supportsFormatCombination(int32_t pos, | ^~~~~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:59:10: error: ‘void nvinfer1::plugin::GridSample3DPlugin::configurePlugin(const int, int32_t, const int, int32_t)’ marked ‘override’, but does not override 59 | void configurePlugin(DynamicPluginTensorDesc const in, | ^~~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:64:12: error: ‘size_t nvinfer1::plugin::GridSample3DPlugin::getWorkspaceSize(const int, int32_t, const int, int32_t) const’ marked ‘override’, but does not override 64 | size_t getWorkspaceSize(PluginTensorDesc const inputs, | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:69:13: error: ‘int32_t nvinfer1::plugin::GridSample3DPlugin::enqueue(const int, const int, const void const, void const, void, cudaStream_t)’ marked ‘override’, but does not override 69 | int32_t enqueue(PluginTensorDesc const inputDesc, | ^~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:77:24: error: ‘nvinfer1::DataType nvinfer1::plugin::GridSample3DPlugin::getOutputDataType(int32_t, const nvinfer1::DataType, int32_t) const’ marked ‘override’, but does not override 77 | nvinfer1::DataType getOutputDataType(int32_t index, | ^~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:80:10: error: ‘void nvinfer1::plugin::GridSample3DPlugin::attachToContext(cudnnContext, cublasContext, nvinfer1::IGpuAllocator)’ marked ‘override’, but does not override 80 | void attachToContext(cudnnContext cudnnContext, | ^~~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:84:10: error: ‘void nvinfer1::plugin::GridSample3DPlugin::detachFromContext()’ marked ‘override’, but does not override 84 | void detachFromContext() noexcept override; | ^~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:87:17: error: ‘const char nvinfer1::plugin::GridSample3DPlugin::getPluginType() const’ marked ‘override’, but does not override 87 | const char getPluginType() const noexcept override; | ^~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:89:17: error: ‘const char nvinfer1::plugin::GridSample3DPlugin::getPluginVersion() const’ marked ‘override’, but does not override 89 | const char getPluginVersion() const noexcept override; | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:91:13: error: ‘int32_t nvinfer1::plugin::GridSample3DPlugin::getNbOutputs() const’ marked ‘override’, but does not override 91 | int32_t getNbOutputs() const noexcept override; | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:93:13: error: ‘int32_t nvinfer1::plugin::GridSample3DPlugin::initialize()’ marked ‘override’, but does not override 93 | int32_t initialize() noexcept override; | ^~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:95:10: error: ‘void nvinfer1::plugin::GridSample3DPlugin::terminate()’ marked ‘override’, but does not override 95 | void terminate() noexcept override; | ^~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:97:12: error: ‘size_t nvinfer1::plugin::GridSample3DPlugin::getSerializationSize() const’ marked ‘override’, but does not override 97 | size_t getSerializationSize() const noexcept override; | ^~~~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:99:10: error: ‘void nvinfer1::plugin::GridSample3DPlugin::serialize(void) const’ marked ‘override’, but does not override 99 | void serialize(void buffer) const noexcept override; | ^~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:101:10: error: ‘void nvinfer1::plugin::GridSample3DPlugin::destroy()’ marked ‘override’, but does not override 101 | void destroy() noexcept override; | ^~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:103:10: error: ‘void nvinfer1::plugin::GridSample3DPlugin::setPluginNamespace(const char)’ marked ‘override’, but does not override 103 | void setPluginNamespace(const char pluginNamespace) noexcept override; | ^~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.h:105:17: error: ‘const char nvinfer1::plugin::GridSample3DPlugin::getPluginNamespace() const’ marked ‘override’, but does not override 105 | const char getPluginNamespace() const noexcept override; | ^~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.cpp:104:1: error: ‘IPluginV2DynamicExt’ does not name a type 104 | IPluginV2DynamicExt GridSample3DPlugin::clone() const noexcept { | ^~~~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.cpp:121:1: error: ‘DimsExprs’ does not name a type 121 | DimsExprs GridSample3DPlugin::getOutputDimensions(int32_t outputIndex, | ^~~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.cpp:139:52: error: ‘PluginTensorDesc’ has not been declared 139 | PluginTensorDesc const inOut, | ^~~~ /data_group/gatkadmin/ai/torch2trt/grid-sample3d-trt-plugin/src/grid_sample_3d_plugin.cpp: In

SeanWangJS commented 6 months ago

你用的 tensorrt版本是多少?我看错误信息里面好多类不识别,8.6.1以上应该不会出现这样的问题。

zmtttt commented 1 month ago

您好。想问一下,up有出现过使用onnx-->engine时,在计算gridsample算子的时候,GPU爆掉的问题呢?请问该怎么解决 @SeanWangJS