Skip to content

Commit 605a9a1

Browse files
authored
Fix integration test gpu_arch_type field (#2060)
All tests in experiments are broken due to the `gpu_arch_type` field added in #2018.
1 parent c8ebd7a commit 605a9a1

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/integration_tests/run_tests.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,10 @@ def run_tests(args, test_list: list[OverrideDefinitions]):
8989
continue
9090

9191
# Skip the test for ROCm
92-
if args.gpu_arch_type == "rocm" and test_flavor.skip_rocm_test:
92+
if (
93+
getattr(args, "gpu_arch_type", "cuda") == "rocm"
94+
and test_flavor.skip_rocm_test
95+
):
9396
continue
9497

9598
# Check if we have enough GPUs

0 commit comments

Comments
 (0)