NVIDIA / NVIDIA/cuda-python

[FEA]: Ability to provide an extra path for cuda-pathfinder to search with high priority

Aperta
#1,054 7 commenti 1 reazione 1 assegnatario Vedi su GitHub

@rwgk ci sta già lavorando.

Dal 30/9/2025.

awaiting-response cuda.pathfinder feature
Lingua principale
Cython
Stelle
3.4k
Fork
329
Merge medio
1g 23h
PR unite (30g)
116

Descrizione

Is this a duplicate?
Area

cuda.pathfinder

Is your feature request related to a problem? Please describe.

The cuda-pathfinder's search path prioritizes libraries from wheels. For example, libnvshmem_host.so comes from libnvidia-nvshmem-cuXX wheel. If I'm working from a development branch of nvshmem, I may need to manually copy libraries over the site-packages directory in order to get my custom libraries to be opened. LD_LIBRARY_PATH is much lower in the priority order (on purpose), so that won't work either.

Describe the solution you'd like

A few different ideas come to mind:

  • <LIBNAME_UPPER_CASE>_PATH_OVERRIDE so each library has an ability to be overridden independently
  • CUDA_PATHFINDER_TOP_PRIORITY_PATH which applies to all pathfinder searches
  • Adding a <library>_HOME environment variable, and prioritizing it above other search paths
Describe alternatives you've considered

I think I listed all 3 above. The current hacky WAR is to copy .so files around on the system.

Additional context

No response

Guida per i contributori

Apri la guida per i contributori

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.