Revert D14613517: [pytorch][PR] Updating onnxtrt submodule to master …

…branch Differential Revision: D14613517 Original commit changeset: dd20d718db55 fbshipit-source-id: d6267ddfc339d04f182e2de1750a601c8d6bf8c6
artsobolev · Mar 27, 2019 · 66e8c74 · 66e8c74
1 parent e912632
commit 66e8c74
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -57,7 +57,7 @@
 	url = https://github.com/onnx/onnx.git
 [submodule "third_party/onnx-tensorrt"]
 	path = third_party/onnx-tensorrt
-	url = https://github.com/onnx/onnx-tensorrt
+	url = https://github.com/bddppq/onnx-tensorrt
 [submodule "third_party/sleef"]
 	path = third_party/sleef
 	url = https://github.com/zdevito/sleef

diff --git a/caffe2/contrib/tensorrt/tensorrt_tranformer.cc b/caffe2/contrib/tensorrt/tensorrt_tranformer.cc
@@ -486,7 +486,7 @@ void TensorRTTransformer::Transform(
   auto trt_builder = tensorrt::TrtObject(nvinfer1::createInferBuilder(logger));
   auto trt_network = tensorrt::TrtObject(trt_builder->createNetwork());
   auto importer =
-      tensorrt::TrtObject(nvonnxparser::createParser(*trt_network, logger));
+      tensorrt::TrtObject(nvonnxparser::createParser(trt_network.get(), logger));
 
   // function to tell whether TensorRT supports a given C2 op or not
   auto supports =

diff --git a/caffe2/contrib/tensorrt/trt_utils.cc b/caffe2/contrib/tensorrt/trt_utils.cc
@@ -13,7 +13,7 @@ std::shared_ptr<nvinfer1::ICudaEngine> BuildTrtEngine(
   auto trt_builder = TrtObject(nvinfer1::createInferBuilder(*logger));
   auto trt_network = TrtObject(trt_builder->createNetwork());
   auto trt_parser =
-      TrtObject(nvonnxparser::createParser(*trt_network, *logger));
+      TrtObject(nvonnxparser::createParser(trt_network.get(), *logger));
   auto status = trt_parser->parse(onnx_model_str.data(), onnx_model_str.size());
   if (!status) {
     const auto num_errors = trt_parser->getNbErrors();

diff --git a/third_party/onnx-tensorrt b/third_party/onnx-tensorrt
+24 −85		CMakeLists.txt
+15 −156		ModelImporter.cpp
+2 −6		ModelImporter.hpp
+6 −27		NvOnnxParser.h
+0 −31		NvOnnxParserTypedefs.h
+2 −8		README.md
+121 −406		builtin_op_importers.cpp
+0 −119		common.hpp
+0 −155		getSupportedAPITest.cpp
+94 −14		main.cpp
+10 −144		onnx2trt_utils.hpp
+5 −106		onnx_backend_test.py
+2 −0		onnx_tensorrt/__init__.py
+12 −52		onnx_tensorrt/backend.py
+17 −57		onnx_tensorrt/tensorrt_engine.py
+53 −14		onnx_trt_backend.cpp
+1 −1		third_party/onnx