minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

fix: restore CPU flash_attn test to use sgl_kernel directly (#22573)

Browse Source 
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Baizhou Zhang
2026-04-10 21:39:20 -07:00
committed by GitHub
co-authored by Claude Opus 4.6
parent f2af00d05a
commit 3c46ff2ac5
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/srt/cpu/test_flash_attn.py
+3 -1
View File
@@ -1,12 +1,14 @@
import unittest import unittest
import sgl_kernel # noqa: F401
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
from utils import parametrize, precision from utils import parametrize, precision
from sglang.jit_kernel.flash_attention import flash_attn_varlen_func
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
flash_attn_varlen_func = torch.ops.sgl_kernel.flash_attn_varlen_func
torch.manual_seed(1234) torch.manual_seed(1234)