ܙddZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ej dkrdZd Znej d krd Zd Znd Zd ZdZdZddZdZdZddZddZdS)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.soc<t}|dj}|S)N libdevice)rinfo)dpathss 7lib/python3.11/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_dir)platformstatic)rr _dllnamepatternr_staticnamepatternmax)librrdir_typelibdir candidates namepatterns r get_cudalibr&,s f}}',H&0HH+1D''}!!(+0#vHHHJ(.C$$OK( ?3z???kC.??rcHt|}tj|S)N)r&ctypesCDLLr!paths r open_cudalibr,>s s  D ;t  rct|d}tj|st |ddS)NTrz not found)r&osr+isfileFileNotFoundErrorr*s rcheck_static_libr2CsK s4 ( ( (D 7>>$  54 3 3 344455rc|dkrtdjS|dkrtdjS|rdnd}t|jS)Nrr rr)rby)r!rr"s r_get_source_variabler5Isb f}}'**   ,//+1D''}),,rTc d} t\}}d|}td|dtd|tddt||\}}td|n+#t$r}td |d }Yd }~nd }~wwxYwd } | D]} t | |}td | t| |rtd|n-tdtj ||d tj fvr_ tddt| td#t$r!}td| d|d }Yd }~d }~wwxYwd} t | |d }td | t| d |rtd|n-tdtj | t!| n.#t"$r!}td| d|d }Yd }~nd }~wwxYwtd} td| tddt%}|rtdntdd }| 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 z: cudadevrtr.z ERROR: failed to find r zFinding libdevice from z ERROR: can't open libdevice)rjoinprintrrsplitr&formatr5r/r+basenamesysrr,OSErrorr2r1r) _platform print_pathsfaileddlloaderr$ locationsdllr+elibsr!wheres rtestrISs}F 799*IIj))  ????@@@ (h(())) 'U3333*55 T )4))****  444555    D3 ** "))#/CC/H/HIIJJJ  6 .$ ' ' ' ' *bg..t44 5 5 5 s|, , , 0e<<<<S!!!f    333BCCC  - C sId 3 3 3D  % %c+?GK,M,M,M N NOOO2 nd#### j"'**400111  333:;;; ! - -E +E + +,,, $%0000 ??D  f  .///:sHBB B.B))B./F  F5F00F5I I;I66I;)NF)F)NT)__doc__r/r>r(numba.misc.findlibrnumba.cuda.cuda_pathsrnumba.cuda.cudadrv.driverrrnumba.cuda.cudadrv.errorrrrrrrr&r,r2r5rIrrrPsC   ''''''000000KKKKKKKK555555<7O!\X#O" O"  @@@@$ 555 ----??????r