diff --git a/tests/ut/test_platform.py b/tests/ut/test_platform.py index 6b088ae7..6a2ce5c8 100644 --- a/tests/ut/test_platform.py +++ b/tests/ut/test_platform.py @@ -490,7 +490,7 @@ class TestNPUPlatform(TestBase): dtype="float16", kv_cache_dtype="float16", block_size=64, - #use_sfa=False, + use_sparse=False, use_mla=True, ) self.assertEqual(result, @@ -503,7 +503,7 @@ class TestNPUPlatform(TestBase): dtype="float16", kv_cache_dtype="float16", block_size=64, - #use_sfa=False, + use_sparse=False, use_mla=False, ) self.assertEqual(