EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Super tiny improve FA3 import error message (#11590)

Browse Source
This commit is contained in:
fzyzcjy
2025-10-15 13:06:31 +08:00
committed by GitHub
parent 91fc5bb5a9
commit 32803fb279
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sgl-kernel/python/sgl_kernel/flash_attn.py
View File

@@ -6,7 +6,9 @@ import torch
try: try:
from sgl_kernel import flash_ops from sgl_kernel import flash_ops
except: except:
raise ImportError("Can not import sgl_kernel. Please check your installation.") raise ImportError(
"Can not import FA3 in sgl_kernel. Please check your installation."
)
try: try:
from ._fa4_interface import flash_attn_varlen_func as flash_attn_varlen_func_v4 from ._fa4_interface import flash_attn_varlen_func as flash_attn_varlen_func_v4