1@j%dZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ej dkrdZd Znej d krd Zd Znd Zd ZdZdZddZdZdZddZdZdS)aCUDA Toolkit libraries lookup utilities. CUDA Toolkit libraries can be available via either: - the `cuda-nvcc` and `cuda-nvrtc` conda packages for CUDA 12, - the `cudatoolkit` conda package for CUDA 11, - 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.soc<t}|dj}|S)N libdevice)rinfo)dpathss HC:\Projects\Historical\venv\Lib\site-packages\numba/cuda/cudadrv/libs.py get_libdevicer"sA kN E Lcttd5}|cdddS#1swxYwYdS)Nrb)openrread)bcfiles ropen_libdevicer(s moot $ ${{}}s>AAFc|dkr#tdjp tdzS|rdnd}t|j}t|||}|rtnt}|rt |n||zS)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_typelibdir candidates namepatterns r get_cudalibr&-s f}}',H&0HH+1D''}!!(+0#vf555J(.C$$OK( ?3z???kC.??rcHt|}tj|S)N)r&ctypesCDLL)r!paths r open_cudalibr+?s s  D ;t  rchtj|st|ddS)Nz not found)osr*isfileFileNotFoundError)r*s rcheck_static_libr0Ds9 7>>$  54 3 3 344455rc|dkrtdjS|dkrtdjS|rdnd}t|jS)Nrr rr)rby)r!rr"s r_get_source_variabler3Isb f}}'**   ,//+1D''}),,rc d} t\}}td|D]}td|td|tddt||\}}tdtd |n+#t$r}td |d }Yd }~nd }~wwxYwtjd kr|st j}t j t jj d|d} t|5} | } d d d n #1swxYwYtd| D} td| D]}td|n##t$rtd|dYnwxYwd} | D]} t!| }td| t%| td| tddt'| td#t$r!}td| d|d }Yd }~d }~wwxYwd} t!| d }td| t%| d td| tddt)|tdn.#t*$r!}td| d|d }Yd }~nd }~wwxYwt%d}td |t-}td| tddt)|tdn.#t*$r!}td| d|d }Yd }~nd }~wwxYw| 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: TNlinuxprocmapsc3"K|] }d|v|V dS)z libcuda.soN).0ss r ztest..s,II!|q7H7HA7H7H7H7HIIrz Mapped libcuda.so paths:z z ERROR: Could not open z) to determine absolute path to libcuda.soznvvm nvrtc cudartzFinding {} from {}z Located atz Trying to open libraryz ERROR: failed to open z: cudadevrtrz Checking libraryz ERROR: failed to find r zFinding libdevice from )rprintrrsysplatformr-getpidr*joinseprrsetsplitOSErrorr&formatr3r+r0r/r)faileddlloaderr$locationdllr*epidmapsfilefr9 locationslibsr!wheres rtestrUSsF 799* /000" # #H /x// " " " " (h(())) 'U3333*55 T f  '''((((  444555 |wvikk7<< VXvFF )h 1vvxx               IItzz||IIIIII . / / /% ) )'X''(((( ) ( ( ( 'X''' ( ( ( ( (  (  $ $ & &D  3 "))#/CC/H/HIIJJJ nd###  ,% 8 8 8 8     &MMMM    EsssAA> ? ? ?FFFFFF  C s4 ( ( (D  % %c+?GK,M,M,M N NOOO .$ ".... f  333:;;; ! - -E +E + +,,, ??D .$ ".... f  333:;;;:sBB B;B66B;F#%E: F#E  F# E F##GG./I J (JJ "/L L=L88L=/N00 O:OO)F)__doc__r-rAr(numba.misc.findlibrnumba.cuda.cuda_pathsrnumba.cuda.cudadrv.driverrrnumba.cuda.cudadrv.errorrrBrrrrr&r+r0r3rUr;rrr[sC   ''''''000000KKKKKKKK555555<7O!\X#O" O"  @@@@$ 555 ----]]]]]r