U -e(@sdZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ej dkrddZd Znej d krxd Zd Znd Zd ZddZddZdddZddZddZdddZddZdS)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 X/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/numba/cuda/cudadrv/libs.py get_libdevice!s rc Cs,ttd}|W5QRSQRXdS)Nrb)openrread)Zbcfiler r ropen_libdevice'srFcCsb|dkrtdjptdS|r$dnd}t|j}t|||d}|rJtnt}|rZt|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_dirstatic)rr _dllnamepatternr_staticnamepatternmax)librdir_typeZlibdir candidatesZ namepatternr r r get_cudalib,s   rcCst|}t|S)N)rctypesCDLL)rpathr r r open_cudalib>sr#cCstj|st|ddS)Nz not found)osr"isfileFileNotFoundError)r"r r rcheck_static_libCs r'cCsD|dkrtdjS|dkr(tdjS|r0dnd}t|jSdS)Nrr rr)rZby)rrrr r r_get_source_variableHs    r(c Csd}zlt\}}td|D]}td|qtd|tdddt||\}}tdtd |Wn4tk r}ztd |d }W5d }~XYnXtjd krX|sXt}tj tjj d|d}z t |} | } W5QRXWn&t k rtd|dYn:Xtdd| D} td| D]}td|qBd} | D]} t| }td| t| td|z tdddt| tdWn8t k r}ztd| |fd }W5d }~XYnXqdd} t| d d}td| t| d dtd|z tdddt|tdWn8tk rv}ztd| |fd }W5d }~XYnXtd}td |t}td|z tdddt|tdWn8tk r}ztd| |fd }W5d }~XYnX| S)!z:Test library lookup. Path info is printed to stdout. FzFinding driver from candidates: z Using loader z Trying to load driverz...)endz okz Loaded from z ERROR: failed to open driver: TNlinuxprocmapsz ERROR: Could not open z) to determine absolute path to libcuda.socss|]}d|kr|VqdS)z libcuda.soNr ).0sr r r ~sztest..z Mapped libcuda.so paths:z znvvm nvrtc cudartzFinding {} from {}z Located atz Trying to open libraryz ERROR: failed to open %s: %sZ cudadevrtrz Checking libraryz ERROR: failed to find %s: %sr zFinding libdevice from )rprintrrsysplatformr$getpidr"joinseprrOSErrorsetsplitrformatr(r#r'r&r)failedZdlloaderrlocationZdllr"epidZmapsfilefr- locationsZlibsrwherer r rtestRs~               rB)F)F)__doc__r$r2r Znumba.misc.findlibrZnumba.cuda.cuda_pathsrZnumba.cuda.cudadrv.driverrrZnumba.cuda.cudadrv.errorrr3rrrrrr#r'r(rBr r r rs,