^j UdZddlZddlZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZddlmZmZdd lmZeGd d ZeGd d ZeegedzfZde dedededzfdZdededzdedzfdZdededzfdZdededzfdZdededzfdZdedededzfdZdededzfdZdededzfdZdededzfdZ eefZ!e"edfe#d <e fZ$e"edfe#d!<ded"e"edfdedzfd#Z%y)$aComposable search steps for locating NVIDIA libraries. Each find step is a callable with signature:: (SearchContext) -> FindResult | None Find steps locate a library file on disk without loading it. The orchestrator in :mod:`load_nvidia_dynamic_lib` handles loading, the already-loaded check, and dependency resolution. Step sequences are defined per search strategy so that adding a new step or strategy only requires adding a function and a tuple entry. This module is intentionally platform-agnostic: it does not branch on the current operating system. Platform differences are routed through the :data:`~cuda.pathfinder._dynamic_libs.search_platform.PLATFORM` instance. N)Callable) dataclassfield)NoReturncast) LibDescriptor)DynamicLibNotFoundError)PLATFORMSearchPlatform)get_cuda_path_or_homec&eZdZUdZeed<eed<y) FindResultz0A library file located on disk (not yet loaded).abs_path found_viaN)__name__ __module__ __qualname____doc__str__annotations__u/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/cuda/pathfinder/_dynamic_libs/search_steps.pyrr&s:MNrrceZdZUdZeed<eZeed<e e Z e e ed<e e Z e e ed<ede fdZede fd Zdefd Zy ) SearchContextz4Mutable state accumulated during the search cascade.descplatform)default_factoryerror_messages attachmentsreturnc.|jjSN)rnameselfs rlibnamezSearchContext.libname7syy~~rchtt|jj|jSr#)rrrlib_searched_forr'r%s rr)zSearchContext.lib_searched_for;s"C77 EFFrcdj|j}dj|j}td|jd|d|)Nz,  zFailure finding "z": )joinrr r r))r&erratts rraise_not_foundzSearchContext.raise_not_found?sVii++,ii(()%(9$:O:O9PPSTWSXXZ[^Z_&`aarN)rrrrrrr rr rlistrrr propertyr'r)rr/rrrrr.s> 'Hn' %d ;NDI;"48Kc8 G#GGbbrrrr anchor_pointr!c :|j|}|D]}ttjtjj ||D]D}tjj |s#tjj|ccSy)zVFind the library directory under *anchor_point* using the descriptor's relative paths.N)anchor_rel_dirssortedglobospathr,isdirnormpath)rrr2rel_dirsrel_pathdirnames r_find_lib_dir_using_anchorr>Isw''-H1dii \8(LMN 1Gww}}W%ww''00 11 rctxlib_dirc |yttdz|jj||j|j |j |jS)z0Find a library file in a resolved lib directory.N)rrrfind_in_lib_dirr'r)rr )r?r@s r_find_using_lib_dirrCSsS  d  $$  KK     OO    rresolved_lib_pathctjj|}tjj|}|dvrtjj|}tjj|}tjj|dk(rtjj|S|Sy)zDerive CTK root from Linux canary path. Supports: - ``$CTK_ROOT/lib64/libfoo.so.*`` - ``$CTK_ROOT/lib/libfoo.so.*`` - ``$CTK_ROOT/targets//lib64/libfoo.so.*`` - ``$CTK_ROOT/targets//lib/libfoo.so.*`` )lib64libtargetsN)r7r8r=basename)rDr@rIparent grandparents r_derive_ctk_root_linuxrLcsggoo/0Gww(H##)ggoof- 77  K (I 577??;/ / rc,ddl}|j|}|j|j}|dk(rE|j|}|j|jdk(r|j|Sy|dk(r|j|Sy)zDerive CTK root from Windows canary path. Supports: - ``$CTK_ROOT/bin/x64/foo.dll`` (CTK 13 style) - ``$CTK_ROOT/bin/foo.dll`` (CTK 12 style) rNx64bin)ntpathr=rIlower)rDrPr@rIrJs r_derive_ctk_root_windowsrRwsnn./Gw'--/H5( ??6 " ( ( *e 3>>&) )  U ~~g&& rc6t|}||St|S)z4Derive CTK root from a resolved canary library path.)rLrR)rDctk_roots rderive_ctk_rootrUs$%&78H #$5 66rrTczt|j|j|}t||}|yt |dS)z3Find a library under a previously derived CTK root.Nzsystem-ctk-root)r>rrrCr)r?rTr@rs rfind_via_ctk_rootrWs;(3<<JG"30H h 1 22rc|jj|j}|sy|jj||j|j |j }| t|dSy)z#Search pip wheel install locations.Nz site-packages)rsite_packages_rel_dirsrfind_in_site_packagesr)rr r)r?r;rs rrZrZsd||22388rrCr)r? conda_prefixanchorr@rs r find_in_condarcsf::>>.1L  \\ , ,\ :F(3<<HG"30H(G,, rct}|yt|j|j|}t ||}| t |dSy)aSearch ``$CUDA_PATH`` / ``$CUDA_HOME``. On Windows, this is the normal fallback for system-installed CTK DLLs when they are not already discoverable via the native ``LoadLibraryExW(..., 0)`` path used by :func:`cuda.pathfinder._dynamic_libs.load_dl_windows.load_with_system_search`. Python 3.8+ does not include ``PATH`` in that native DLL search. The returned ``found_via`` is always ``"CUDA_PATH"`` regardless of which environment variable actually provided the value. N CUDA_PATH)r r>rrrCr)r? cuda_homer@rs rfind_in_cuda_pathrgsM&'I(3<<KG"30H(K00 r.EARLY_FIND_STEPSLATE_FIND_STEPSstepsc.|D]}||}||cSy)z1Run find steps in order, returning the first hit.Nr)r?rjstepresults rrun_find_stepsrns*c  M r)&rr6r7collections.abcr dataclassesrrtypingrr,cuda.pathfinder._dynamic_libs.lib_descriptorr,cuda.pathfinder._dynamic_libs.load_dl_commonr -cuda.pathfinder._dynamic_libs.search_platformr r cuda.pathfinder._utils.env_varsr rrFindSteprr>rCrLrRrUrWrZrcrgrhtuplerrirnrrrrxs$ $(!FPRA   bb b. ]OZ$%66 7]n\_dgjndn ] S4Z C$J  ccDj(d (7s7sTz73=3C3J4+@)O%# &O*;(<x}%< eHcM.BzTXGXr