Skip to content

Commit 4367ef1

Browse files
Make RNG deterministic in KV cache unit test (#4738)
1 parent dcd3dff commit 4367ef1

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

op_tests/triton_tests/test_kv_cache.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,8 @@ def test_cat_and_cache_mla(
219219
):
220220
pytest.skip("FP4 KV cache is only supported in GFX950 and GFX1250")
221221

222+
torch.manual_seed(0)
223+
222224
dtype = torch.bfloat16
223225
k_lora = torch.randn((T, KH, D_lora), dtype=torch.float32, device="cuda") / (
224226
20 if cache_dtype != torch.bfloat16 else 1

0 commit comments

Comments
 (0)