Enable arg_ops_test/unique_ops_test on AMD/rocm (pytorch#16853)

Summary: Verified both tests are passing on rocm 2.1 env. Pull Request resolved: pytorch#16853 Differential Revision: D13996279 Pulled By: bddppq fbshipit-source-id: c0df610d7d9ca8d80ed2d1339cdadef59105a71c
strisunshinewentingwang · Feb 8, 2019 · c65b03b · c65b03b
1 parent bca358a
commit c65b03b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.jenkins/caffe2/test.sh b/.jenkins/caffe2/test.sh
@@ -91,10 +91,8 @@ if [[ $BUILD_ENVIRONMENT == *-rocm* ]]; then
   # Currently these tests are failing on ROCM platform:
 
   # Unknown reasons, need to debug
-  rocm_ignore_test+=("--ignore $caffe2_pypath/python/operator_test/arg_ops_test.py")
   rocm_ignore_test+=("--ignore $caffe2_pypath/python/operator_test/piecewise_linear_transform_test.py")
   rocm_ignore_test+=("--ignore $caffe2_pypath/python/operator_test/softmax_ops_test.py")
-  rocm_ignore_test+=("--ignore $caffe2_pypath/python/operator_test/unique_ops_test.py")
 
   # On ROCm, RCCL (distributed) development isn't complete.
   # https://github.com/ROCmSoftwarePlatform/rccl

diff --git a/caffe2/python/operator_test/unique_ops_test.py b/caffe2/python/operator_test/unique_ops_test.py
@@ -67,3 +67,7 @@ def test_unique_op(self, X, return_remapping, gc, dc):
             inputs=[X],
             reference=partial(_unique_ref, return_inverse=return_remapping),
         )
+
+if __name__ == "__main__":
+    import unittest
+    unittest.main()