Skip to content

Commit b910a6b

Browse files
committed
Pull out **Environment variables** into an added section, after manual inspection of cuda_paths.py. Minor additional edits.
1 parent 5a835d7 commit b910a6b

File tree

1 file changed

+9
-7
lines changed
  • cuda_bindings/cuda/bindings/_path_finder

1 file changed

+9
-7
lines changed

cuda_bindings/cuda/bindings/_path_finder/README.md

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,13 @@ strategy for locating NVIDIA shared libraries:
2828
- Scans `sys.path` to find libraries installed via NVIDIA Python wheels.
2929

3030
2. **Conda Environments**
31-
- Leverages Conda-specific paths through our fork of `get_cuda_paths()` from Numba.
31+
- Leverages Conda-specific paths through our fork of `get_cuda_paths()`
32+
from numba-cuda.
3233

33-
3. **System Installations**
34+
3. **Environment variables**
35+
- Relies on `CUDA_HOME`/`CUDA_PATH` environment variables if set.
36+
37+
4. **System Installations**
3438
- Checks traditional system locations through these paths:
3539
- Linux: `/usr/local/cuda/lib64`
3640
- Windows: `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vX.Y\bin`
@@ -39,10 +43,8 @@ strategy for locating NVIDIA shared libraries:
3943
- Versioned CUDA directories like `/usr/local/cuda-12.3`
4044
- Distribution-specific packages (RPM/DEB)
4145
EXCEPT Debian's `nvidia-cuda-toolkit`
42-
- Relies on `CUDA_HOME`/`CUDA_PATH` environment variables if set, but falls
43-
back to hardcoded paths when unset
4446

45-
4. **OS Default Mechanisms**
47+
5. **OS Default Mechanisms**
4648
- Falls back to native loader:
4749
- `dlopen()` on Linux
4850
- `LoadLibraryW()` on Windows
@@ -54,8 +56,8 @@ mechanism that ensures all libraries are found in the same way.
5456

5557
The current implementation balances stability and evolution:
5658

57-
- **Baseline Foundation:** Uses a fork of Numba's `cuda_paths.py` that has been
58-
battle-tested in production environments
59+
- **Baseline Foundation:** Uses a fork of numba-cuda's `cuda_paths.py` that has been
60+
battle-tested in production environments.
5961

6062
- **Validation Infrastructure:** Comprehensive CI testing matrix being developed to cover:
6163
- Various Linux/Windows environments

0 commit comments

Comments
 (0)