fix UT

Signed-off-by: zzhx1 <[email protected]>

Author: zzhx1

tests/ut/attention/test_mla_v1.py

@@ -649,9 +649,10 @@ def test_build_for_graph_capture_prefill(self, mock_distributed):
             str(ctx.exception))
 
 
-class TestAscendMLAImpl(TestBase):
+class TestAscendMLAImpl:
 
-    def setUp(self):
+    @pytest.fixture(autouse=True)
+    def setup(self, mock_distributed):
         num_heads = 256
         head_size = 1024
         scale = 0.1