DeformableConvRT.h fixes ,cmake cuda arch auto detection

2021-09-07 23:57:52 +05:30
parent 03473743c4
commit 18cc6abbb6
4 changed files with 82 additions and 11 deletions
@@ -110,6 +110,6 @@ namespace nvinfer1 {
        static std::vector<PluginField> mPluginAttributes;
        std::string mPluginNamespace;
    };
-
+    REGISTER_TENSORRT_PLUGIN(DeformableConvRTPluginCreator);
 };
 #endif