Merge with cnet

2021-05-04 11:50:08 +02:00
parent 10f39d1055 d1ae1791d9
commit f8327e2dac
4 changed files with 29 additions and 17 deletions
@@ -140,7 +140,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
        engineRT = builderRT->buildEngineWithConfig(*networkRT, *configRT);
 #else 
        //engineRT = builderRT->buildCudaEngine(*networkRT);
-        engineRT = std::shared_ptr<nvinfer1::ICudaEngine>(builderRT->buildCudaEngine(*networkRT));
+        engineRT = builderRT->buildCudaEngine(*networkRT);
 #endif
        if(engineRT == nullptr)
            FatalError("cloud not build cuda engine")