Unverified Commit 7b743ba9 authored by zhanqiuhu's avatar zhanqiuhu Committed by GitHub
Browse files

[CI] Fix: pass string cache_dtype in test_register_kv_caches (#38836)

parent 188defbd
......@@ -1613,7 +1613,7 @@ def test_register_kv_caches(
)
]
],
cache_dtype=torch.bfloat16,
cache_dtype="bfloat16",
device=torch.accelerator.current_device_index(),
kernel_block_sizes=[block_size],
)
......
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