Run ONNX cuda backend test cases via ROCm
authorbddppq <bai@in.tum.de>
Thu, 13 Dec 2018 23:07:10 +0000 (15:07 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 13 Dec 2018 23:10:00 +0000 (15:10 -0800)
commit855d9e1f19d69e5b3963a2ec7ac3cf0fc31120d9
treeef06ae4ce427c1913cc72ecdb54c93a188884bdc
parent6911ce19d7fcf06e7af241e6494b23acdc320dc4
Run ONNX cuda backend test cases via ROCm

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15069

Differential Revision: D13427757

Pulled By: bddppq

fbshipit-source-id: ba0273d75986cd5b146f7041a83c63ddf9c6c0cf
caffe2/python/onnx/backend.py
caffe2/python/onnx/tests/onnx_backend_test.py