fix: use triton_attn as default vision attention on B300 (SM103) (#25570)

2026-05-19 11:00:07 +08:00
parent 31e324391b
commit d8e66e54e5
1 changed files with 1 additions and 1 deletions
@@ -951,7 +951,7 @@ class VisionAttention(nn.Module):
            major, minor = get_device_capability()
            if major == 9:
                backend = "fa3"
-            elif major == 10:
+            elif major == 10 and minor != 3:
                backend = "fa4"
            else:
                backend = "triton_attn"