Skip to content

[FEA]: Support LoadedDL.from_distribution #759

@brandon-b-miller

Description

@brandon-b-miller

Is this a duplicate?

Area

cuda.bindings

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

Numba would like to report to the user not only the path it found it's dependencies on, but also the distribution that they came from. For instance consider the output of numba -s:

CUDA Libraries Test Output:                                                                                                                                                                                                                               
Finding driver from candidates:                                                                                                                                                                                                                           
        libcuda.so                                                                                                                                                                                                                                        
        libcuda.so.1                                                                                                                                                                                                                                      
        /usr/lib/libcuda.so                                                                                                                                                                                                                               
        /usr/lib/libcuda.so.1                                                                                                                                                                                                                             
        /usr/lib64/libcuda.so                                                                                                                                                                                                                             
        /usr/lib64/libcuda.so.1                                                                                                                                                                                                                           
Using loader <class 'ctypes.CDLL'>                                                                                                                                                                                                                        
        Trying to load driver...        ok                                                                                                                                                                                                                
                Loaded from libcuda.so                                                                                                                                                                                                                    
        Mapped libcuda.so paths:                                                                                                                                                                                                                          
                /usr/lib/x86_64-linux-gnu/libcuda.so.535.161.08                                                                                                                                                                                           
Finding nvvm from Conda environment (NVIDIA package)                                                                                                                                                                                                      
        Located at /mambaforge/envs/myenv/nvvm/lib64/libnvvm.so.4.0.0                                                                                                                                                                    
        Trying to open library...       ok                                                                                                                                                                                                                
Finding nvrtc from Conda environment (NVIDIA package)                                                                                                                                                                                                     
        Located at /mambaforge/envs/myenv/lib/libnvrtc.so.12.9.86                                                                                                                                                                        
        Trying to open library...       ok                                                                                                                                                                                                                
Finding cudart from Conda environment (NVIDIA package)                                                                                                                                                                                                    
        Located at /mambaforge/envs/myenv/lib/libcudart.so.12.9.79                                                                                                                                                                       
        Trying to open library...       ok                                                                                                                                                                                                                
Finding cudadevrt from Conda environment (NVIDIA package)                                                                                                                                                                                                 
        Located at /mambaforge/envs/myenv/lib/libcudadevrt.a                                                                                                                                                                             
        Checking library...     ok                                                                                                                                                                                                                        
Finding libdevice from Conda environment (NVIDIA package)                                                                                                                                                                                                 
        Located at /mambaforge/envs/myenv/nvvm/libdevice/libdevice.10.bc                                                                                                                                                                 
        Checking library...     ok                                                                                                                                                                                                                        
Include directory configuration variable:                                                                                                                                                                                                                 
        CUDA_INCLUDE_PATH=/usr/local/cuda/include                                                                                                                                                                                                         
Finding include directory from Conda environment (NVIDIA package)                                                                                                                                                                                         
        Located at /mambaforge/envs/myenv/targets/x86_64-linux/include                                                                                                                                                                   
        Checking include directory...   ok  

Describe the solution you'd like

It would be useful if the LoadedDL class supported a from_distribution property that reported the source, such as wheel, conda or system ctk.

Describe alternatives you've considered

Without this, using pathfinder would require decoding the source distribution from the path supplied by pathfinder if we want to maintain the information currently supplied by the output.

Additional context

No response

Metadata

Metadata

Assignees

Labels

cuda.pathfinderEverything related to the cuda.pathfinder modulefeatureNew feature or requesttriageNeeds the team's attention

Type

No type

Projects

Status

Todo

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions