o 8ήcT@sdZddlZddlZddlZddlmZddlmZejdkr$dZ 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-target-vg8gfxp4/lib/python/numba/cuda/cudadrv/libs.py get_libdevices r cCs8ttd }|WdS1swYdS)Nrb)openr read)bcfiler r r open_libdevice$s$rFcCsX|dkrtdjp tdStdj}t||||d}|r tnt}|r(t|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)librrlibdir candidates namepatternr r r get_cudalib)s   rcCst|}t|S)N)rctypesCDLLrpathr r r open_cudalib:s r"cCs*t|dd}tj|st|ddS)NTrz not found)rosr!isfileFileNotFoundErrorr r r r check_static_lib?s  r'cCs4|dkr tdjS|dkrtdjStdjS)Nrrr)rby)rr r r _get_source_variableEs    r)Tc Csd}d}|D]V}t||}td|t||r!td|n tdtj||dtj fvr^ztddd t |td Wqt y]}ztd ||fd }WYd}~qd}~wwqd }t||d d}td|t||rztd|n tdtj|zt |Wnt y}ztd||fd }WYd}~nd}~wwtd}td|tddd t}|rtd | Stdd }| 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: %sT cudadevrtr#z ERROR: failed to find %s: %srzFinding libdevice from z ERROR: can't open libdevice)splitrprintformatr)r$r!basenamesysrr"OSErrorr'r&r ) _platform print_pathsfailedlibsrr!ewherer r r testNsT        r8)NF)NT)__doc__r$r0rnumba.misc.findlibrnumba.cuda.cuda_pathsrrrrr rrr"r'r)r8r r r r s*