create a base framework for adding tests

tests/L0/common_utils.py

+'''
+This file contains common utility functions for running the unit tests on ROCM.
+'''
+
+import torch
+
+TEST_WITH_ROCM = os.getenv('APEX_TEST_WITH_ROCM', '0') == '1'
+
+## Wrapper to skip the unit tests.
+def skipIfRocm(fn):
+    @wraps(fn)
+    def wrapper(*args, **kwargs):
+        if TEST_WITH_ROCM:
+            raise unittest.SkipTest("test doesn't currently work on ROCm stack.")
+        else:
+            fn(*args, **kwargs)
+    return wrapper

tests/L0/run_test.py

 import unittest
 import sys
+from common_utils import *
+
 
 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',
+    'run_pyprof_data',
+    'run_mlp'
+]
+
 runner = unittest.TextTestRunner(verbosity=2)
 
 errcode = 0
 
 for test_dir in test_dirs:
+    if test_dir in ROCM_BLACKLIST:
+        continue
     suite = unittest.TestLoader().discover(test_dir)
 
     print("\nExecuting tests from " + test_dir)