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

This commit is contained in:
Mohammad Miadh Angkad
2026-07-14 09:01:23 -07:00
committed by GitHub
parent 04af94d150
commit 271e5ef5c3
@@ -26,6 +26,7 @@ from unittest import mock
import torch
from sglang.kernels.ops.attention import flash_mla_sm120 as fmod
from sglang.kernels.ops.attention.flash_mla_sm120 import (
_D,
_NOPE_DIM,
@@ -43,7 +44,6 @@ from sglang.kernels.ops.attention.flash_mla_sm120_triton import (
_merge_partial_attn,
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.test_utils import CustomTestCase