[Bugfix] Preserve benchmark metric sample counts in results (#7624)

Signed-off-by: psv666 <2693925048@qq.com>
Co-authored-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
psv666andamy-why-3459 authored and GitHub committed 2026-09-16 08:39:01 -07:00
1 parent 538eddd74d
commit 1053f7b0bd
3 files changed
+92 -6

No files matched your search

+10 -6
View File
@@ -570,6 +570,7 @@ def test_omni_tpot_baseline_accepts_measured_finite_sample():
@pytest.mark.parametrize(
("num_tpot_samples", "mean_tpot_ms", "match"),
[
(None, 10.0, "no measurable TPOT samples"),
(0, float("nan"), "no measurable TPOT samples"),
(1, float("nan"), "mean_tpot_ms is not finite"),
],
@@ -577,14 +578,17 @@ def test_omni_tpot_baseline_accepts_measured_finite_sample():
def test_omni_tpot_baseline_rejects_missing_or_nonfinite_sample(num_tpot_samples, mean_tpot_ms, match):
from tests.dfx.perf.scripts.run_benchmark import assert_result
result = {
"completed": 1,
"Hardware": "H100",
"mean_tpot_ms": mean_tpot_ms,
}
if num_tpot_samples is not None:
result["num_tpot_samples"] = num_tpot_samples
with pytest.raises(AssertionError, match=match):
assert_result(
{
"completed": 1,
"Hardware": "H100",
"num_tpot_samples": num_tpot_samples,
"mean_tpot_ms": mean_tpot_ms,
},
result,
{"baseline": {"H100": {"mean_tpot_ms": 20.0}}},
1,
)