Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
eb45455
FEA Allow using cuML benchmark tools on systems without cuML installed
dantegd Dec 10, 2025
028932e
FIX style fixes and a few code cleanups
dantegd Dec 10, 2025
6c0ea2d
FIX style fixes and a few code cleanups
dantegd Dec 10, 2025
023b601
ENH Address PR review feedback
dantegd Feb 13, 2026
5e9e181
ENH code cleanup and implement python -m cuml.benchmark functionality
dantegd Feb 13, 2026
aaa7265
FIX code cleanups
dantegd Feb 13, 2026
ed18641
FIX resolve coderabbit feedback
dantegd Feb 24, 2026
cc9f72e
Fixup style checks.
csadorf Mar 2, 2026
b8a912d
ignore run_benchmark.py E402 error
csadorf Mar 2, 2026
a4f17e2
Document required dependencies in stand-alone mode
csadorf Mar 2, 2026
4bf54ce
Do not exit when importung run_benchmark.py
csadorf Mar 2, 2026
d2bc333
Simplify module structure.
csadorf Mar 2, 2026
57e1ba2
Fixup the dual-import logic.
csadorf Mar 2, 2026
16c9385
Fix the is_gpu_available function
csadorf Mar 2, 2026
91aa6e2
Guard sys.exit() in __main__
csadorf Mar 3, 2026
bacc598
Replace HAS_CUML global with is_cuml_available function.
csadorf Mar 3, 2026
f5e5515
Refactor parser definition into cli module.
csadorf Mar 3, 2026
ce126be
Refactor run_benchmark function into multiple smaller functions.
csadorf Mar 3, 2026
0036dcd
Split the sentence on running the benchmark suite into separate parag…
csadorf Mar 3, 2026
5efad69
Cache the all_algorithms function
csadorf Mar 3, 2026
5995341
Clean up temporary directories after use.
csadorf Mar 3, 2026
7b6c67c
Fix _build_optimized_fil_classifier function
csadorf Mar 3, 2026
e6ba5ed
Use perf_counter() instead of time.time()
csadorf Mar 3, 2026
7aa965d
Fix BenchmarkTimer doc-string
csadorf Mar 3, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion python/cuml/cuml/benchmark/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,10 @@
#
# SPDX-FileCopyrightText: Copyright (c) 2019, NVIDIA CORPORATION.
# SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION.
# SPDX-License-Identifier: Apache-2.0
#
"""cuML benchmark package.

Run benchmarks via:
- Full mode: python -m cuml.benchmark [options] [algorithms...]
- Standalone: python run_benchmark.py [options] [algorithms...] (from benchmark dir)
"""
12 changes: 12 additions & 0 deletions python/cuml/cuml/benchmark/__main__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
#
# SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION.
# SPDX-License-Identifier: Apache-2.0
#
"""Entry point for python -m cuml.benchmark (full mode, requires cuML)."""

import sys

from cuml.benchmark.run_benchmarks import main

if __name__ == "__main__":
sys.exit(main())
Loading