NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expand README

Browse Source 
Make it explicit in README that we build and locally install NVBench first,
and then build Python package use the library as a dependency.

The nvbench library is installed into Python layout alongside the native
extension.
This commit is contained in:
Oleksandr Pavlyk
2025-07-24 10:39:15 -05:00
parent 5e8c17c740
commit 445d881eda
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/README.md
View File

@@ -19,10 +19,18 @@ Now switch to python folder, configure and install NVBench library, and install
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
```
### Build Python extension
Specify location local installation of `NVBench` library and perform editable `pip install`:
```
nvbench_DIR=$(pwd)/nvbench_install/lib/cmake CUDACXX=/usr/local/cuda/bin/nvcc pip install -e .
```
Note that `CUDACXX` must be set for NVBench cmake script to work, but Python extension itself only uses host compiler.
### Verify that package works
```