Skip to content

GitLab

Unverified Commit d46d417b authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[CI Perf] Only test bfloat16 for tests/compile/test_fusion_all_reduce.py (#23132) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent 0167efe2
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/compile/test_fusion_all_reduce.py
View file @ d46d417b
......@@ -148,7 +148,7 @@ class TestAllReduceFusedAddRMSNormStaticQuantFP4Model(torch.nn.Module):
@pytest.mark.parametrize("batch_size", [8])
@pytest.mark.parametrize("seq_len", [8])
@pytest.mark.parametrize("hidden_size", [16])
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16])
@pytest.mark.parametrize("dtype", [torch.bfloat16])
@pytest.mark.skipif(envs.VLLM_TARGET_DEVICE not in ["cuda"],
reason="Only test on CUDA")
@pytest.mark.skipif(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment