[NPU][DSV4]dsv4 enable cpp (#39820)

Co-authored-by: Even Zhou <even.y.zhou@outlook.com>
2026-09-20 09:17:27 +08:00
parent 9cc7da2ab0
commit d2f291c934
3 changed files with 11 additions and 9 deletions
@@ -712,7 +712,7 @@ class TestArch35SparseAttentionDispatch(unittest.TestCase):
            kwargs = _sparse_attn_kv_quant_kwargs()

        self.assertIs(metadata_op, custom_ops.npu_sparse_attn_sharedkv_metadata)
-        self.assertIs(attention_op, npu_ops.sparse_attn_sharedkv)
+        self.assertIs(attention_op, custom_ops.npu_sparse_attn_sharedkv)
        self.assertEqual(kwargs, {})