[CI] Fix UT (#5106)
Fix broken ut introduced by #5053
- vLLM version: v0.12.0
- vLLM main:
ad32e3e19c
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
@@ -490,7 +490,7 @@ class TestNPUPlatform(TestBase):
|
|||||||
dtype="float16",
|
dtype="float16",
|
||||||
kv_cache_dtype="float16",
|
kv_cache_dtype="float16",
|
||||||
block_size=64,
|
block_size=64,
|
||||||
#use_sfa=False,
|
use_sparse=False,
|
||||||
use_mla=True,
|
use_mla=True,
|
||||||
)
|
)
|
||||||
self.assertEqual(result,
|
self.assertEqual(result,
|
||||||
@@ -503,7 +503,7 @@ class TestNPUPlatform(TestBase):
|
|||||||
dtype="float16",
|
dtype="float16",
|
||||||
kv_cache_dtype="float16",
|
kv_cache_dtype="float16",
|
||||||
block_size=64,
|
block_size=64,
|
||||||
#use_sfa=False,
|
use_sparse=False,
|
||||||
use_mla=False,
|
use_mla=False,
|
||||||
)
|
)
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
|
|||||||
Reference in New Issue
Block a user