1
import unittest
2
import sys
3

4
test_dirs = ["run_amp", "run_fp16util", "run_optimizers", "run_fused_layer_norm", "run_pyprof_nvtx", "run_pyprof_data", "run_mlp"]
5

6
runner = unittest.TextTestRunner(verbosity=2)
7

8
errcode = 0
9

10
for test_dir in test_dirs:
11
    suite = unittest.TestLoader().discover(test_dir)
12

13
    print("\nExecuting tests from " + test_dir)
14

15
    result = runner.run(suite)
16

17
    if not result.wasSuccessful():
18
        errcode = 1
19

20
sys.exit(errcode)
21

22