frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Back out "Revert D14613517: [pytorch][PR] Updating onnxtrt submodule to master branch" (#18514)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18514

Original commit changeset: d6267ddfc339

Reviewed By: bddppq

Differential Revision: D14634476

fbshipit-source-id: 2633b0b4c512d71001e5c20cd79c0c0d7856f942
This commit is contained in:
Li Yu
2019-03-26 23:41:35 -07:00
committed by Facebook Github Bot
parent eae7ad4ca8
commit a3933b87c6
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.gitmodules vendored
View File

@ -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/bddppq/onnx-tensorrt
url = https://github.com/onnx/onnx-tensorrt
[submodule "third_party/sleef"]
path = third_party/sleef
url = https://github.com/zdevito/sleef

2
caffe2/contrib/tensorrt/tensorrt_tranformer.cc
View File

@ -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.get(), logger));
tensorrt::TrtObject(nvonnxparser::createParser(*trt_network, logger));
// function to tell whether TensorRT supports a given C2 op or not
auto supports =

2
caffe2/contrib/tensorrt/trt_utils.cc
View File

@ -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.get(), *logger));
TrtObject(nvonnxparser::createParser(*trt_network, *logger));
auto status = trt_parser->parse(onnx_model_str.data(), onnx_model_str.size());
if (!status) {
const auto num_errors = trt_parser->getNbErrors();

2
third_party/onnx-tensorrt vendored

Submodule third_party/onnx-tensorrt updated: 9884f286a2...cb3d8066f2