Kernels community fa3 (#20796)

2026-04-07 12:48:44 -07:00
parent 0c204fbd57
commit 1a8eb890f6
20 changed files with 1956 additions and 61 deletions
@@ -1,15 +1,12 @@
 import unittest

-import sgl_kernel  # noqa: F401
 import torch
 import torch.nn.functional as F
 from utils import parametrize, precision

+from sglang.jit_kernel.flash_attention import flash_attn_varlen_func
 from sglang.test.test_utils import CustomTestCase

-flash_attn_varlen_func = torch.ops.sgl_kernel.flash_attn_varlen_func
-
-
 torch.manual_seed(1234)