Super tiny improve FA3 import error message (#11590)

2025-10-15 13:06:31 +08:00
parent 91fc5bb5a9
commit 32803fb279
1 changed files with 3 additions and 1 deletions
--- a/sgl-kernel/python/sgl_kernel/flash_attn.py
+++ b/sgl-kernel/python/sgl_kernel/flash_attn.py
@@ -6,7 +6,9 @@ import torch
 try:
    from sgl_kernel import flash_ops
 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:
    from ._fa4_interface import flash_attn_varlen_func as flash_attn_varlen_func_v4