;j %SSKJr SSKrSSKrSSKrSSKJr SSKJr SSK J r SSK J r SSK Jr SSKJrJrJr SS KJr SS KJr \ (aSS KJr \"S S 55r\S/\S-4rSSjrSSjrSSjrSSjrSSjrSSjrSSjr SSjr!\\\\ \!4r"S\#S'S Sjr$\RJS!Sj5r&S"Sjr'g)#) annotationsN)Callable) dataclass) TYPE_CHECKING)-_resolve_system_loaded_abs_path_in_subprocess)derive_ctk_root)HEADER_DESCRIPTORSplatform_include_subdirsresolve_conda_anchor)get_cuda_path_or_home)find_sub_dirs_all_sitepackages)HeaderDescriptorc4\rSrSr%S\S'S\S'S SjrSrg) LocatedHeaderDir! str | Noneabs_pathstr found_viac8[UR5UlgN) _abs_normr)selfs }/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/cuda/pathfinder/_headers/find_nvidia_headers.py __post_init__LocatedHeaderDir.__post_init__&s!$--0 )rN)returnNone)__name__ __module__ __qualname____firstlineno____annotations__r__static_attributes__rrrr!sN1rrrcU(a<[RR[RRU55$gr)ospathnormpathabspath)r)s rrr2s* ww 566 rcz[RR[RRX55$r)r(r)isfilejoin)dirpathbasenames r_joined_isfiler18s" 77>>"'',,w9 ::rc6URnURH}n[RR X5n[ U5H8n[RR XE5n[ Xb5(dM4Us s $ [ XB5(dM{Us $ g)zNSearch for a header under *anchor_point* using the descriptor's layout fields.N)header_basenameanchor_include_rel_dirsr(r)r.r r1)desc anchor_point h_basenamerel_diridirsubdircdirs r_locate_in_anchor_layoutr<<sw%%J//ww||L2.t4F77<<-Dd// 5 $ + +K0 rcURHUn[[URS555H+n[ X R 5(dM[ USS9s s $ MW g)z#Search pip wheel install locations./z site-packagesrrN)site_packages_dirsr tuplesplitr1r3r)r5sub_dirhdr_dirs rfind_in_site_packagesrEOsS**5eGMM#>.1L ';L0D):gNN rcP[5nUcg[X5nUb [USS9$g)z'Search ``$CUDA_PATH`` / ``$CUDA_HOME``.N CUDA_PATHr?)r r<r)r5 cuda_homeresults rfind_in_cuda_pathrRgs3%'I %d 6F ;GG rcUR(dg[S5nUcg[U5nUcg[X5nUb [ USS9$g)a8Try CTK header lookup via CTK-root canary probing. Skips immediately if the descriptor does not opt in (``use_ctk_root_canary``). Otherwise, system-loads ``cudart`` in a fully isolated Python subprocess, derives CTK root from the resolved library path, and searches the expected include layout under that root. Ncudartzsystem-ctk-rootr?)use_ctk_root_canaryrrr<r)r5canary_abs_pathctk_rootrQs rfind_via_ctk_root_canaryrXrsW  # #CHMO/H %d 5F ;LMM rcURHPn[[R"U5SS9H+n[X R5(dM[ USS9s s $ MR g)z2Search system install directories (glob patterns).T)reversesupported_install_dirr?N)system_install_dirssortedglobr1r3r)r5patternrDs rfind_in_system_install_dirsr`sQ++dii0$?Gg';';<<'D[\\@, rtuple[HeaderFindStep, ...] FIND_STEPSc4UHnU"U5nUcMUs $ g)z1Run find steps in order, returning the first hit.Nr&)r5stepssteprQs rrun_find_stepsrfs%d  M rcr[R"U5nUc[SU<35e[U[5$)uLocate the header directory for a supported NVIDIA library. Args: libname (str): The short name of the library whose headers are needed (e.g., ``"nvrtc"``, ``"cusolver"``, ``"nvshmem"``). Returns: LocatedHeaderDir or None: A LocatedHeaderDir object containing the absolute path to the discovered header directory and information about where it was found, or ``None`` if the headers cannot be found. Raises: RuntimeError: If ``libname`` is not in the supported set. Search order: 1. **NVIDIA Python wheels** — site-packages directories from the descriptor. 2. **Conda environments** — platform-specific conda include layouts. 3. **CUDA Toolkit environment variables** — ``CUDA_PATH`` / ``CUDA_HOME``. 4. **CTK root canary probe** — subprocess canary (descriptors with ``use_ctk_root_canary=True`` only). 5. **System install directories** — glob patterns from the descriptor. zUNKNOWN libname=)r rJ RuntimeErrorrfrb)libnamer5s rlocate_nvidia_header_directoryrjs90  ! !' *D |.gZ011 $ ++rcB[U5nU(a UR$S$)uLLocate the header directory for a supported NVIDIA library. Args: libname (str): The short name of the library whose headers are needed (e.g., ``"nvrtc"``, ``"cusolver"``, ``"nvshmem"``). Returns: str or None: Absolute path to the discovered header directory, or ``None`` if the headers cannot be found. Raises: RuntimeError: If ``libname`` is not in the supported set. Search order: 1. **NVIDIA Python wheels** — site-packages directories from the descriptor. 2. **Conda environments** — platform-specific conda include layouts. 3. **CUDA Toolkit environment variables** — ``CUDA_PATH`` / ``CUDA_HOME``. 4. **CTK root canary probe** — subprocess canary (descriptors with ``use_ctk_root_canary=True`` only). 5. **System install directories** — glob patterns from the descriptor. N)rjr)rifounds rfind_nvidia_header_directoryrms, +7 3E"5>>,,r)r)rrr)r/rr0rrbool)r5rr6rrr)r5rrLocatedHeaderDir | None)r5rrdrarro)rirrro)rirrr)( __future__r functoolsr^r(collections.abcr dataclassesrtypingr5cuda.pathfinder._dynamic_libs.load_nvidia_dynamic_libr*cuda.pathfinder._dynamic_libs.search_stepsr*cuda.pathfinder._headers.header_descriptorr r r cuda.pathfinder._utils.env_varsr $cuda.pathfinder._utils.find_sub_dirsr rrHeaderFindSteprr1r<rErMrRrXr`rbr$rfcacherjrmr&rrr|s# $! G BOK 11 1-.0@40GGH ; & , * & ,,:-r