diff --git a/build_tools/github_actions/configure_ci.py b/build_tools/github_actions/configure_ci.py index 86978ce4fcd0..7a62b1d633c5 100755 --- a/build_tools/github_actions/configure_ci.py +++ b/build_tools/github_actions/configure_ci.py @@ -191,6 +191,10 @@ def contains(cls, val): ".github/worklflows/ci_windows_x64_msvc.yml", ], ), + ( + "test_torch", + ["tests/external/iree-test-suites/torch*"], + ), ] PR_DESCRIPTION_TEMPLATE = string.Template("${title}\n\n${body}") diff --git a/tests/external/iree-test-suites/torch_ops/torch_ops_gpu_hip_gfx1201_O3.json b/tests/external/iree-test-suites/torch_ops/torch_ops_gpu_hip_gfx1201_O3.json index a1dd3b1519cf..baebf6555755 100644 --- a/tests/external/iree-test-suites/torch_ops/torch_ops_gpu_hip_gfx1201_O3.json +++ b/tests/external/iree-test-suites/torch_ops/torch_ops_gpu_hip_gfx1201_O3.json @@ -8,12 +8,29 @@ "iree_run_module_flags": [ "--device=hip" ], - "skip_compile_tests": [], + "skip_compile_tests": [ + "InterestingShapesBiasAdd/997x997xi8_NN_bias" + ], "skip_run_tests": [ "ABPlusC/64x64xf16", - "ATB/64x64xf16" + "ATB/64x64xf16", + "AB/1024x1024xf32_bench", + "AB/128x128xf32_bench", + "AB/2048x2048xf32_bench", + "AB/256x256xf32_bench", + "AB/4096x4096xf32_bench", + "AB/512x512xf32_bench", + "AB/8192x8192xf32_bench" ], "expected_compile_failures": [], "expected_run_failures": [], - "golden_times_ms": {} + "golden_times_ms": { + "AB/1024x1024xf32_bench" : 0.0, + "AB/128x128xf32_bench" : 0.0, + "AB/2048x2048xf32_bench" : 0.0, + "AB/256x256xf32_bench" : 0.0, + "AB/4096x4096xf32_bench" : 0.0, + "AB/512x512xf32_bench" : 0.0, + "AB/8192x8192xf32_bench" : 0.0 + } }