Enable working ROCm tests (#19169)
authorJohannes M Dieterich <johannes.dieterich@amd.com>
Sat, 13 Apr 2019 04:42:10 +0000 (21:42 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 13 Apr 2019 04:51:10 +0000 (21:51 -0700)
commitd8669a2c7eca95dd1a9b7180a4d873fde40bbf9b
tree463bab5b4e423817aabff369b6b42a3749e9b6aa
parentca02558d4006678b38cdd1a4df30576c9eefb9c5
Enable working ROCm tests (#19169)

Summary:
Enable multi-GPU tests that work with ROCm 2.2. Have been run three times on CI to ensure stability.

While there, remove skipIfRocm annotations for tests that depend on MAGMA. They still skip but now for the correct reason (no MAGMA) to improve our diagnostics.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19169

Differential Revision: D14924812

Pulled By: bddppq

fbshipit-source-id: 8b88f58bba58a08ddcd439e899a0abc6198fef64
test/test_cuda.py
test/test_nn.py