[CPU][CI]: fix a few issues that cause XEON CI failures (#36413)
Co-authored-by: Zhang, Mingxu <mingxu.zhang@intel.com> Co-authored-by: Claude <noreply@anthropic.com> Co-authored-by: MingxuZh <109504044+MingxuZh@users.noreply.github.com>
This commit is contained in:
co-authored by
Zhang, Mingxu
Claude
MingxuZh
parent
ea48cb04cc
commit
b294bd4bc7
@@ -34,7 +34,7 @@ class TestIntelAMXAttnBackend(CustomTestCase):
|
||||
return DEFAULT_MLA_MODEL_NAME_FOR_TEST
|
||||
|
||||
@intel_amx_benchmark(
|
||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.2"],
|
||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.3"],
|
||||
min_throughput=40,
|
||||
)
|
||||
def test_latency_default_model(self):
|
||||
@@ -84,7 +84,7 @@ class TestDPAttention(CustomTestCase):
|
||||
"--attention-backend",
|
||||
"intel_amx",
|
||||
"--mem-fraction-static",
|
||||
"0.2",
|
||||
"0.3",
|
||||
"--disable-overlap-schedule",
|
||||
"--tp",
|
||||
"2",
|
||||
|
||||
@@ -20,14 +20,14 @@ register_cpu_ci(est_time=47, suite="base-b-test-cpu")
|
||||
class TestIntelAMXAttnBackendQuant(CustomTestCase):
|
||||
|
||||
@intel_amx_benchmark(
|
||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.2"],
|
||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.3"],
|
||||
min_throughput=150,
|
||||
)
|
||||
def test_latency_fp8_qwen(self):
|
||||
return DEFAULT_MODEL_NAME_FOR_TEST_QWEN_FP8
|
||||
|
||||
@intel_amx_benchmark(
|
||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.2"],
|
||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.3"],
|
||||
min_throughput=50,
|
||||
)
|
||||
def test_latency_fp8_moe_model(self):
|
||||
|
||||
Reference in New Issue
Block a user