xKgudZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ejdk(rdZd Znejd k(rd Zd Znd Zd ZdZdZddZdZdZddZdZy)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.soc8t}|dj}|S)N libdevice)rinfo)dpathss [/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/cudadrv/libs.py get_libdevicer"sA kN  E Lcvttd5}|jcdddS#1swYyxYw)Nrb)openrread)bcfiles ropen_libdevicer(s% mot ${{} % $ $s/8c|dk(r"tdjxs tdzS|rdnd}t|j}t|||}|rtnt}|r t |S||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&-st f}',,H&0HH+1'}!(+00#vf5J(.$OK(3z??kC.??rcBt|}tj|S)N)r&ctypesCDLL)r!paths r open_cudalibr+?s s D ;;t rc^tjj|st|dy)Nz not found)osr*isfileFileNotFoundError)r*s rcheck_static_libr0Ds) 77>>$ 4& 344 rc|dk(rtdjS|dk(rtdjS|rdnd}t|jS)Nrr rr)rby)r!rr"s r_get_source_variabler3IsW f}'***  ,///+1'}),,,rc d} t\}}td|D]}td|td|tddt||\}}tdtd |tj d k(r|st j}t jjt jjd|d} t|5} | j} d d d td jD} td| D]}td| dj} | D]a} t!| }tdj#| t%| td| tddt'| tdcd} t!| d }tdj#| t%| d td| tddt)|tdt%d}td |t-}td| tddt)|td| S#t$r}td |d }Yd }~d }~wwxYw#1swYxYw#t$rtd|dYwxYw#t$r}td| d|d }Yd }~d }~wwxYw#t*$r}td| d|d }Yd }~d }~wwxYw#t*$r}td| d|d }Yd }~| Sd }~wwxYw)!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|vs| yw)z libcuda.soN).0ss r ztest..sI|!|q7HA|s z 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 79* /0"H Bxj/ "# hZ() 'U3*5 T f   '( ||wviik77<< VuvF )h1vvx Itzz|III . /%XJ'(&  $ $ &D3 "))#/CC/HIJ nd#  ,% 8   &M C s4 (D  % %c+?GK,M NO .$ ". f ! -E #E7 +, ?D .$ ". f :[  045   ( ,XJ7'' (  (2  sA> ?F   3:;  3:;: sA0I/ J#$J5J##K6#K, #L/ J8JJJ J##J?>J? K) K$$K), L5L  L L=L88L=)F)__doc__r-rAr(numba.misc.findlibrnumba.cuda.cuda_pathsrnumba.cuda.cudadrv.driverrrnumba.cuda.cudadrv.errorrrBrrrrr&r+r0r3rUr;rrr[s  '0K5<<7O!\\X#O" O"  @$ 5 -]r