diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index fe04608fd3bab06049f9e1e2639ba13b1166732d..0e554491015218acb0593c8c8562d29c29c68bc4 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -20,7 +20,7 @@ ops_test: stage: ops_test script: - if [ -z "$TARGET_SOCS" ]; then TARGET_SOCS=random; fi - - python tools/bazel_adb_run.py --target="//mace/ops:ops_test" --run_target=True --stdout_processor=ops_test_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS + - python tools/bazel_adb_run.py --target="//mace/ops:ops_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS ops_benchmark: stage: ops_benchmark @@ -32,5 +32,5 @@ extra_tests: stage: extra_tests script: - if [ -z "$TARGET_SOCS" ]; then TARGET_SOCS=random; fi - - python tools/bazel_adb_run.py --target="//mace/kernels:kernels_test" --run_target=True --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS - - python tools/bazel_adb_run.py --target="//mace/utils:tuner_test" --run_target=True --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS + - python tools/bazel_adb_run.py --target="//mace/kernels:kernels_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS + - python tools/bazel_adb_run.py --target="//mace/utils:tuner_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS diff --git a/tools/bazel_adb_run.py b/tools/bazel_adb_run.py index 88d8c1814b84aaa36e178f2c844ce3675a279d58..d1d322087e0d73ad16f5b99f3575a390df77db29 100644 --- a/tools/bazel_adb_run.py +++ b/tools/bazel_adb_run.py @@ -19,7 +19,7 @@ def stdout_processor(stdout, device_properties, abi): pass -def ops_test_stdout_processor(stdout, device_properties, abi): +def unittest_stdout_processor(stdout, device_properties, abi): stdout_lines = stdout.split("\n") for line in stdout_lines: if "Aborted" in line or "FAILED" in line: