Skip to content

GitLab

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: ...@@ -17,7 +17,7 @@ def is_fa3_supported(device=None) -> bool:
# https://docs.nvidia.com/cuda/cuda-c-programming-guide/#shared-memory-8-x # https://docs.nvidia.com/cuda/cuda-c-programming-guide/#shared-memory-8-x
# now sgl-kernel only build fa3 for sm90a && cuda >= 12.4 # now sgl-kernel only build fa3 for sm90a && cuda >= 12.4
return ( return (
(torch.cuda.get_device_capability(device)[0] >= 9) (torch.cuda.get_device_capability(device)[0] == 9)
and (torch.version.cuda >= "12.4") and (torch.version.cuda >= "12.4")
# or torch.cuda.get_device_capability(device) == (8, 0) # or torch.cuda.get_device_capability(device) == (8, 0)
# or torch.cuda.get_device_capability(device) == (8, 7) # 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