f16 test

Signed-off-by: NickLucche <[email protected]>
NickLucche · Jan 31, 2025 · 80b94da · 80b94da
1 parent d93bf4d
commit 80b94da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/spec_decode/e2e/test_integration_dist_tp2.py b/tests/spec_decode/e2e/test_integration_dist_tp2.py
@@ -78,7 +78,7 @@ def test_target_model_tp_gt_1(common_llm_kwargs, per_test_common_llm_kwargs,
 
         # precision
         "--dtype",
-        "bfloat16",
+        "float16",
     ]])
 @pytest.mark.parametrize("per_test_common_llm_kwargs", [[]])
 @pytest.mark.parametrize("baseline_llm_kwargs", [[]])