mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-05-03 13:11:14 +00:00
6552ef503c
The prototype is based on pybind11 to minimize boiler-plate code needed to deal with move-only semantics of many nvbench classes.
22 lines
584 B
Markdown
22 lines
584 B
Markdown
# CUDA Kernel Benchmarking Package
|
|
|
|
This package provides Python API to CUDA Kernel Benchmarking Library `NVBench`.
|
|
|
|
## Building
|
|
|
|
### Build `NVBench` project
|
|
|
|
```
|
|
cd nvbench/python
|
|
cmake -B nvbench_build --preset nvbench-ci -S $(pwd)/.. -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc -DNVBench_ENABLE_EXAMPLES=OFF -DCMAKE_INSTALL_PREFIX=$(pwd)/nvbench_install
|
|
cmake --build nvbench_build/ --config Release --target install
|
|
|
|
nvbench_DIR=$(pwd)/nvbench_install/lib/cmake CUDACXX=/usr/local/cuda/bin/nvcc pip install -e .
|
|
```
|
|
|
|
### Verify that package works
|
|
|
|
```
|
|
python test/run_1.py
|
|
```
|