U hd@sdZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ej dkrddZd Znej d krxd Zd Znd Zd ZddZddZdddZddZddZdddZd ddZdS)!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_paths)locate_driver_and_loader load_driver)CudaSupportErrorwin32z%s.dllz%s.libdarwinz lib%s.dylibzlib%s.azlib%s.socCst}|dj}|S)N libdevice)rinfo)dpathsr P/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/cuda/cudadrv/libs.py get_libdevice!s rc Cs,ttd}|W5QRSQRXdS)Nrb)openrread)Zbcfiler r ropen_libdevice'srFcCsd|dkrtdjptdS|r$dnd}t|j}t||||d}|rLtnt}|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. nvvmstatic_cudalib_dir cudalib_dir)platformstatic)rr _dllnamepatternr_staticnamepatternmax)librrdir_typeZlibdir candidatesZ namepatternr r r get_cudalib,s   rcCst|}t|S)N)rctypesCDLLrpathr r r open_cudalib>sr$cCs*t|dd}tj|s&t|ddS)NTrz not found)rosr#isfileFileNotFoundErrorr"r r rcheck_static_libCs  r)cCsD|dkrtdjS|dkr(tdjS|r0dnd}t|jSdS)Nrr rr)rZby)rrrr r r_get_source_variableIs    r*Tc Cs0d}z^t\}}d|}td|dtd|tdddt||\}}td|Wn4tk r}ztd |d }W5d }~XYnXd } | D]} t| |}td | t| |rtd|ntdt j ||d t j fkrz tdddt| tdWqtk rN}ztd| |fd }W5d }~XYqXqd} t| |d d}td | t| d d|rtd|ntdt j |z t| Wn8tk r}ztd| |fd }W5d }~XYnXtd} td| tdddt}|rtdn tdd }| S)z:Test library lookup. Path info is printed to stdout. Fz, z Finding driver from candidates: z...z Using loader z trying to load driver)endz ok, loaded from z ERROR: failed to open driver: TNz nvvm cudartzFinding {} from {}z located atz named z trying to open libraryz okz ERROR: failed to open %s: %sZ cudadevrtr%z ERROR: failed to find %s: %sr zFinding libdevice from z ERROR: can't open libdevice)rjoinprintrrsplitrformatr*r&r#basenamesysrr$OSErrorr)r(r) _platformZ print_pathsfailedZdlloaderr locationsZdllr#eZlibsrwherer r rtestSsb             r8)NF)F)NT)__doc__r&r1r Znumba.misc.findlibrZnumba.cuda.cuda_pathsrZnumba.cuda.cudadrv.driverrrZnumba.cuda.cudadrv.errorrrrrrrrr$r)r*r8r r r rs,