NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-07-12 18:17:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Compute quantiles the same way C++ does

Browse Source 
Use rounded-rank method, rather than NumPy's quantile
This commit is contained in:
Oleksandr Pavlyk
2026-06-26 18:12:46 -05:00
parent ad4a76dbe1
commit 02e2d6f097
2 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/test/test_nvbench_compare.py
View File

@@ -801,14 +801,15 @@ def test_compare_gpu_timings_classifies_common_cases(tmp_path, nvbench_compare):
mean=100.0,
stdev=1.0,
stdev_relative=0.01,
sample_values=[4.0, 5.0, 6.0],
sample_values=[4.0, 5.0, 6.0, 100.0],
),
)
assert mixed_robust_summary_and_bulk is not None
assert mixed_robust_summary_and_bulk.ref_time == pytest.approx(10.0)
assert mixed_robust_summary_and_bulk.cmp_time == pytest.approx(5.0)
assert mixed_robust_summary_and_bulk.cmp_time == pytest.approx(6.0)
assert mixed_robust_summary_and_bulk.ref_interval.center == pytest.approx(10.0)
assert mixed_robust_summary_and_bulk.cmp_interval.center == pytest.approx(5.0)
assert mixed_robust_summary_and_bulk.cmp_interval.center == pytest.approx(6.0)
assert mixed_robust_summary_and_bulk.cmp_interval.upper == pytest.approx(6.0)
ref_interval_timing = make_gpu_timing_data(
nvbench_compare,