Skip to content

Fix to make ONNXIFI flow work#9340

Closed
yinghai wants to merge 3 commits intopytorch:masterfrom
yinghai:fix_onnifi
Closed

Fix to make ONNXIFI flow work#9340
yinghai wants to merge 3 commits intopytorch:masterfrom
yinghai:fix_onnifi

Conversation

@yinghai
Copy link
Contributor

@yinghai yinghai commented Jul 11, 2018

Small step to have Relu test work.

Test Plan:

  • Patch TensorRT ONNXIFI interface onnx/onnx-tensorrt#14 and build libtrt_onnxify.so
  • ln -s /home/yinghai/onnx-tensorrt/build/libtrt_onnxify.so ~/lib/libonnxifi.so
  • export LD_LIBRARY_PATH=/home/yinghai/lib:$LD_LIBRARY_PATH
  • Patch this PR and build Pytorch
  • pytest -s test_onnxifi.py

@yinghai yinghai requested review from Maratyszcza and bddppq July 11, 2018 06:00
@yinghai yinghai added the caffe2 label Jul 11, 2018
outputs=[make_tensor_value_info("Y", onnx.TensorProto.FLOAT,
[batch_size, 1, 3, 2])])
model_def = make_model(graph_def, producer_name='relu-test')
op = core.CreateOperator(

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

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.

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

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.

goodlux pushed a commit to goodlux/pytorch that referenced this pull request Aug 15, 2018
Summary:
Small step to have Relu test work.
Pull Request resolved: pytorch#9340

Reviewed By: bddppq

Differential Revision: D8807018

Pulled By: yinghai

fbshipit-source-id: 429f3185e12afb12aaecfea8dd9595fdf838d356
@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.

5 participants