Skip to content

GitLab

"src/array/cuda/spmm_hetero.hip" did not exist on "619d735df5dc2a62eca5a00e11e4290407169cb1"
Unverified Commit c1dd773c authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix: use fa3 unit test on hopper only (#5304)

parent 6f859379
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
sgl-kernel/tests/test_flash_attention.py
View file @ c1dd773c
......@@ -17,7 +17,7 @@ def is_fa3_supported(device=None) -> bool:
# https://docs.nvidia.com/cuda/cuda-c-programming-guide/#shared-memory-8-x
# now sgl-kernel only build fa3 for sm90a && cuda >= 12.4
return (
(torch.cuda.get_device_capability(device)[0] >= 9)
(torch.cuda.get_device_capability(device)[0] == 9)
and (torch.version.cuda >= "12.4")
# or torch.cuda.get_device_capability(device) == (8, 0)
# or torch.cuda.get_device_capability(device) == (8, 7)
......
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