Skip to content

Commit 61c34ae

Browse files
committed
Spawn in itself doesn't seem to work on CUDA
Signed-off-by: Gregory Shtrasberg <[email protected]>
1 parent becbd2f commit 61c34ae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/compile/test_fusion_attn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def fusion_op_check(test_pass_manager: TestPassManager, quant_key: QuantKey,
6565
@pytest.mark.skipif(not current_platform.supports_fp8(), reason="Need FP8")
6666
@pytest.mark.skipif(not current_platform.is_cuda_alike(),
6767
reason="Only test CUDA and ROCm")
68-
@create_new_process_for_each_test("spawn")
68+
@create_new_process_for_each_test()
6969
def test_attention_fusion_v0(example_prompts, monkeypatch, model: str,
7070
quant_key: QuantKey, use_triton_fa: bool):
7171
# Clean Dynamo cache to avoid reusing other test cases
@@ -161,7 +161,7 @@ def test_attention_fusion_v0(example_prompts, monkeypatch, model: str,
161161
@pytest.mark.skipif(not current_platform.supports_fp8(), reason="Need FP8")
162162
@pytest.mark.skipif(not current_platform.is_cuda_alike(),
163163
reason="Only test CUDA and ROCm")
164-
@create_new_process_for_each_test("spawn")
164+
@create_new_process_for_each_test()
165165
def test_attention_fusion_v1(example_prompts, monkeypatch, model: str,
166166
quant_key: QuantKey, use_split_attention: bool):
167167
# Clean Dynamo cache to avoid reusing other test cases

0 commit comments

Comments
 (0)