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

[CI] Fix Flash MLA SM120 test import path (#31199)

Browse Source
This commit is contained in:
Mohammad Miadh Angkad
2026-07-14 09:01:23 -07:00
committed by GitHub
parent 04af94d150
commit 271e5ef5c3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/kernels/test_flash_mla_backends.py
+1 -1
View File
@@ -26,6 +26,7 @@ from unittest import mock
import torch import torch
from sglang.kernels.ops.attention import flash_mla_sm120 as fmod
from sglang.kernels.ops.attention.flash_mla_sm120 import ( from sglang.kernels.ops.attention.flash_mla_sm120 import (
_D, _D,
_NOPE_DIM, _NOPE_DIM,
@@ -43,7 +44,6 @@ from sglang.kernels.ops.attention.flash_mla_sm120_triton import (
_merge_partial_attn, _merge_partial_attn,
flash_mla_sparse_decode_triton, flash_mla_sparse_decode_triton,
) )
from sglang.srt.layers.attention import flash_mla_sm120 as fmod
from sglang.test.ci.ci_register import register_cuda_ci from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase