From 271e5ef5c3ac5733908ce270d556985c1dee0624 Mon Sep 17 00:00:00 2001
From: Mohammad Miadh Angkad <176301910+mmangkad@users.noreply.github.com>
Date: Wed, 15 Jul 2026 00:01:23 +0800
Subject: [PATCH] [CI] Fix Flash MLA SM120 test import path (#31199)

---
 test/registered/kernels/test_flash_mla_backends.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/registered/kernels/test_flash_mla_backends.py b/test/registered/kernels/test_flash_mla_backends.py
index 98dee7bf4..973dbf454 100644
--- a/test/registered/kernels/test_flash_mla_backends.py
+++ b/test/registered/kernels/test_flash_mla_backends.py
@@ -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