[AMD] CI - add partitions for stage-b-test-small-1-gpu-amd (#17345)

2026-01-19 08:16:07 -08:00
parent 2ea02f0642
commit 1a053a810c
5 changed files with 8 additions and 8 deletions
@@ -32,7 +32,7 @@ class TestDeepseekV32DP(CustomTestCase):
            "8",
            "--enable-dp-attention",
            "--model-loader-extra-config",
-            '{"enable_multithread_load": true, "num_threads": 64}',
+            '{"enable_multithread_load": true}',
        ]
        if is_in_amd_ci():
            other_args += [
@@ -101,7 +101,7 @@ class TestDeepseekV32TP(CustomTestCase):
            "--tp",
            "8",
            "--model-loader-extra-config",
-            '{"enable_multithread_load": true, "num_threads": 64}',
+            '{"enable_multithread_load": true}',
        ]
        if is_in_amd_ci():
            other_args += [