Skip to content

Update onnx-tensort module to the latest#9469

Closed
yinghai wants to merge 2 commits intopytorch:masterfrom
yinghai:more_tests
Closed

Update onnx-tensort module to the latest#9469
yinghai wants to merge 2 commits intopytorch:masterfrom
yinghai:more_tests

Conversation

@yinghai
Copy link
Contributor

@yinghai yinghai commented Jul 16, 2018

Update onnx-tensort to follow up recent changes.

Test Plan:

(venv)[yinghai:trt: (more_tests)]$ pytest -v
=========================================================================================== test session starts ============================================================================================
platform linux2 -- Python 2.7.5, pytest-3.4.0, py-1.5.2, pluggy-0.6.0 -- /data/users/yinghai/aml/venv/bin/python
cachedir: ../../../.pytest_cache
rootdir: /data/users/yinghai/aml/pytorch, inifile:
plugins: cov-2.5.1, nbval-0.9.0, hypothesis-3.44.23
collected 12 items

test_trt.py::TensorRTOpTest::test_bvlc_alexnet PASSED                                                                                                                                                [  8%]
test_trt.py::TensorRTOpTest::test_densenet121 PASSED                                                                                                                                                 [ 16%]
test_trt.py::TensorRTOpTest::test_inception_v1 PASSED                                                                                                                                                [ 25%]
test_trt.py::TensorRTOpTest::test_inception_v2 PASSED                                                                                                                                                [ 33%]
test_trt.py::TensorRTOpTest::test_relu_graph_big_batch PASSED                                                                                                                                        [ 41%]
test_trt.py::TensorRTOpTest::test_relu_graph_simple PASSED                                                                                                                                           [ 50%]
test_trt.py::TensorRTOpTest::test_resnet50 PASSED                                                                                                                                                    [ 58%]
test_trt.py::TensorRTOpTest::test_shufflenet <- ../../../../../usr/lib64/python2.7/unittest/case.py SKIPPED                                                                                          [ 66%]
test_trt.py::TensorRTOpTest::test_squeezenet PASSED                                                                                                                                                  [ 75%]
test_trt.py::TensorRTOpTest::test_vgg16 PASSED                                                                                                                                                       [ 83%]
test_trt.py::TensorRTOpTest::test_vgg19 PASSED                                                                                                                                                       [ 91%]
test_trt.py::TensorRTTransformTest::test_resnet50_core PASSED                                                                                                                                        [100%]

================================================================================== 11 passed, 1 skipped in 231.54 seconds ==================================================================================

@yinghai yinghai requested review from Maratyszcza and bddppq July 16, 2018 18:39
@yinghai yinghai added the caffe2 label Jul 16, 2018

class OnnxifiTest(TestCase):
@unittest.skipIf(not workspace.C.use_trt, "No TensortRT support")
@unittest.skip("Need ONNXIFI backednd support")

This comment was marked as off-topic.

Y = workspace.FetchBlob("Y")
np.testing.assert_almost_equal(Y, np.maximum(X, 0))

@unittest.skip("Need ONNXIFI backednd support")

This comment was marked as off-topic.

Copy link
Contributor

@Maratyszcza Maratyszcza left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@yinghai has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

goldsborough pushed a commit to goldsborough/pytorch that referenced this pull request Jul 20, 2018
Summary:
Update onnx-tensort to follow up recent changes.
Pull Request resolved: pytorch#9469

Reviewed By: Maratyszcza

Differential Revision: D8866704

Pulled By: yinghai

fbshipit-source-id: 3b96ec2fa28470f0d4b5a7c62ab332eeba4bdb12
jramseyer pushed a commit to jramseyer/pytorch that referenced this pull request Jul 30, 2018
Summary:
Update onnx-tensort to follow up recent changes.
Pull Request resolved: pytorch#9469

Reviewed By: Maratyszcza

Differential Revision: D8866704

Pulled By: yinghai

fbshipit-source-id: 3b96ec2fa28470f0d4b5a7c62ab332eeba4bdb12
goodlux pushed a commit to goodlux/pytorch that referenced this pull request Aug 15, 2018
Summary:
Update onnx-tensort to follow up recent changes.
Pull Request resolved: pytorch#9469

Reviewed By: Maratyszcza

Differential Revision: D8866704

Pulled By: yinghai

fbshipit-source-id: 3b96ec2fa28470f0d4b5a7c62ab332eeba4bdb12
@ezyang ezyang added the merged label Jun 26, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants