enable fp16_utils test suite

apex/__init__.py

@@ -18,3 +18,6 @@ from . import fp16_utils
 from . import optimizers
 from . import normalization
 from . import pyprof
+
+#common utilties to run tests on ROCm.
+from . import testing

apex/testing/__init__.py

+#from common_utils import *

tests/L0/common_utils.py → apex/testing/common_utils.py

@@ -3,6 +3,9 @@ This file contains common utility functions for running the unit tests on ROCM.
 '''
 
 import torch
+import os
+import sys
+
 
 TEST_WITH_ROCM = os.getenv('APEX_TEST_WITH_ROCM', '0') == '1'
 

tests/L0/run_test.py

 import unittest
 import sys
-from common_utils import *
 
+from apex.testing.common_utils import TEST_WITH_ROCM, skipIfRocm
 
 test_dirs = ["run_amp", "run_fp16util", "run_optimizers", "run_fused_layer_norm", "run_pyprof_nvtx", "run_pyprof_data", "run_mlp"]
 
 ROCM_BLACKLIST = [
     'run_amp',
-    'run_fp16util',
     'run_optimizers',
     'run_fused_layer_norm',
     'run_pyprof_nvtx',
@@ -20,7 +19,7 @@ runner = unittest.TextTestRunner(verbosity=2)
 errcode = 0
 
 for test_dir in test_dirs:
-    if test_dir in ROCM_BLACKLIST:
+    if (test_dir in ROCM_BLACKLIST) and TEST_WITH_ROCM:
         continue
     suite = unittest.TestLoader().discover(test_dir)