U ,dT@sdZddlZddlZddlZddlmZddlmZejdkrHdZ dZ nejdkr\d Z d Z nd Z d Z d d Z ddZ dddZ ddZddZddZdddZdS)a2Cudatoolkit libraries lookup utilities. Cudatoolkit libraries can be available via either: - the `cudatoolkit` conda package, - a user supplied location from CUDA_HOME, - a system wide location, - package-specific locations (e.g. the Debian NVIDIA packages), - or can be discovered by the system loader. N)find_lib)get_cuda_pathswin32z%s.dllz%s.libdarwinz lib%s.dylibzlib%s.azlib%s.socCst}|dj}|S)N libdevice)rinfo)dpathsr ;/tmp/pip-unpacked-wheel-eu7e0c37/numba/cuda/cudadrv/libs.py get_libdevices r c Cs,ttd}|W5QRSQRXdS)Nrb)openr read)Zbcfiler r r open_libdevice$srFcCsX|dkrtdjptdStdj}t||||d}|r@tnt}|rPt|S||S)a Find the path of a CUDA library based on a search of known locations. If the search fails, return a generic filename for the library (e.g. 'libnvvm.so' for 'nvvm') so that we may attempt to load it using the system loader's search mechanism. nvvm cudalib_dir)platformstatic)rr_dllnamepatternr_staticnamepatternmax)librrZlibdir candidatesZ namepatternr r r get_cudalib)s   rcCst|}t|S)N)rctypesCDLLrpathr r r open_cudalib:srcCs*t|dd}tj|s&t|ddS)NTrz not found)rosrisfileFileNotFoundErrorrr r r check_static_lib?s  r$cCs8|dkrtdjS|dkr(tdjStdjSdS)Nrrr)rZby)rr r r _get_source_variableEs   r%Tc Csd}d}|D]}t||}td|t||rBtd|ntdtj||dtj fkrz tddd t |td Wqt k r}ztd ||fd }W5d}~XYqXqd }t||d d}td|t||rtd|ntdtj|z t |Wn8t k rH}ztd||fd }W5d}~XYnXtd}td|tddd t}|rtd n tdd }| S)z:Test library lookup. Path info is printed to stdout. Fz nvvm cudartzFinding {} from {}z located atz named Nz trying to open libraryz...)endz okz ERROR: failed to open %s: %sTZ cudadevrtr z ERROR: failed to find %s: %srzFinding libdevice from z ERROR: can't open libdevice)splitrprintformatr%r!rbasenamesysrrOSErrorr$r#r ) _platformZ print_pathsfailedZlibsrrewherer r r testNsH        r1)NF)NT)__doc__r!r+rZnumba.misc.findlibrZnumba.cuda.cuda_pathsrrrrr rrrr$r%r1r r r r s(