[CPU] Raise mem-fraction-static 0.1->0.2 in intel_amx backend a/b (#36260)
Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -34,7 +34,7 @@ class TestIntelAMXAttnBackend(CustomTestCase):
|
|||||||
return DEFAULT_MLA_MODEL_NAME_FOR_TEST
|
return DEFAULT_MLA_MODEL_NAME_FOR_TEST
|
||||||
|
|
||||||
@intel_amx_benchmark(
|
@intel_amx_benchmark(
|
||||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.1"],
|
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.2"],
|
||||||
min_throughput=40,
|
min_throughput=40,
|
||||||
)
|
)
|
||||||
def test_latency_default_model(self):
|
def test_latency_default_model(self):
|
||||||
@@ -84,7 +84,7 @@ class TestDPAttention(CustomTestCase):
|
|||||||
"--attention-backend",
|
"--attention-backend",
|
||||||
"intel_amx",
|
"intel_amx",
|
||||||
"--mem-fraction-static",
|
"--mem-fraction-static",
|
||||||
"0.1",
|
"0.2",
|
||||||
"--disable-overlap-schedule",
|
"--disable-overlap-schedule",
|
||||||
"--tp",
|
"--tp",
|
||||||
"2",
|
"2",
|
||||||
|
|||||||
@@ -20,14 +20,14 @@ register_cpu_ci(est_time=47, suite="base-b-test-cpu")
|
|||||||
class TestIntelAMXAttnBackendQuant(CustomTestCase):
|
class TestIntelAMXAttnBackendQuant(CustomTestCase):
|
||||||
|
|
||||||
@intel_amx_benchmark(
|
@intel_amx_benchmark(
|
||||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.1"],
|
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.2"],
|
||||||
min_throughput=150,
|
min_throughput=150,
|
||||||
)
|
)
|
||||||
def test_latency_fp8_qwen(self):
|
def test_latency_fp8_qwen(self):
|
||||||
return DEFAULT_MODEL_NAME_FOR_TEST_QWEN_FP8
|
return DEFAULT_MODEL_NAME_FOR_TEST_QWEN_FP8
|
||||||
|
|
||||||
@intel_amx_benchmark(
|
@intel_amx_benchmark(
|
||||||
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.1"],
|
extra_args=["--batch-size", "4", "--mem-fraction-static", "0.2"],
|
||||||
min_throughput=50,
|
min_throughput=50,
|
||||||
)
|
)
|
||||||
def test_latency_fp8_moe_model(self):
|
def test_latency_fp8_moe_model(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user