a d@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 `/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/numba/cuda/cudadrv/libs.py get_libdevice!s rcCs8ttd}|WdS1s*0YdS)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_typelibdir candidatesZ namepatternr r r get_cudalib,s   r cCst|}t|S)N)r ctypesCDLLrpathr r r open_cudalib>sr%cCs*t|dd}tj|s&t|ddS)NTrz not found)r osr$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 Cs6d}z^t\}}d|}td|dtd|tdddt||\}}td|Wn6ty}ztd |d }WYd }~n d }~00d } | D]} t| |}td | t| |rtd|ntdt j ||d t j fvrz tdddt| tdWqtyR}z td| |fd }WYd }~qd }~00qd} t| |d d}td | t| d d|rtd|ntdt j |z t| Wn:ty}z td| |fd }WYd }~n d }~00td} td| tdddt}|r$tdn 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)rjoinprintrrsplitr formatr+r'r$basenamesysrr%OSErrorr*r)r) _platformZ print_pathsfailedZdlloaderr locationsdllr$eZlibsrwherer r rtestSsb             r:)NF)F)NT)__doc__r'r2r!Znumba.misc.findlibrZnumba.cuda.cuda_pathsrZnumba.cuda.cudadrv.driverrrZnumba.cuda.cudadrv.errorrrrrrrr r%r*r+r:r r r rs,