^j dUddlmZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZmZdd lmZdd lmZe rdd lmZeGd d ZedgedzfZddZddZddZddZddZddZddZ ddZ!eeee e!fZ"de#d<d dZ$ejJd!dZ&d"dZ'y)#) 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)HeaderDescriptorc*eZdZUded<ded<ddZy)LocatedHeaderDir str | Noneabs_pathstr found_viac8t|j|_yN) _abs_normr)selfs w/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/cuda/pathfinder/_headers/find_nvidia_headers.py __post_init__zLocatedHeaderDir.__post_init__&s!$--0 N)returnNone)__name__ __module__ __qualname____annotations__rrrrr!sN1rrrc|r>"'',,w9 ::rc"|j}|jD]t}tjj ||}t |D]4}tjj ||}t ||s0|ccSt ||sr|cSy)zNSearch for a header under *anchor_point* using the descriptor's layout fields.N)header_basenameanchor_include_rel_dirsr$r%r*r r-)desc anchor_point h_basenamerel_diridirsubdircdirs r_locate_in_anchor_layoutr8<s%%J//ww||L'2.t4 F77<<f-DdJ/   $ +K rc|jD]Q}tt|jdD])}t ||j st |dccSSy)z#Search pip wheel install locations./z site-packagesrrN)site_packages_dirsr tuplesplitr-r/r)r1sub_dirhdr_dirs rfind_in_site_packagesrAOs_**U5eGMM#>.1L 'l;L0|D):gNN rcTt}|yt||}| t|dSy)z'Search ``$CUDA_PATH`` / ``$CUDA_HOME``.N CUDA_PATHr;)r r8r)r1 cuda_homeresults rfind_in_cuda_pathrNgs5%'I %dI 6F ;GG rc|jsytd}|yt|}|yt||}| t |dSy)aLTry 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_canaryrrr8r)r1canary_abs_pathctk_rootrMs rfind_via_ctk_root_canaryrTrsY  # #CHMO/H %dH 5F ;LMM rc|jD]N}ttj|dD])}t||jst |dccSPy)z2Search system install directories (glob patterns).T)reversesupported_install_dirr;N)system_install_dirssortedglobr-r/r)r1patternr@s rfind_in_system_install_dirsr\s]++]dii0$? ]Ggt';';<'D[\\ ]] rtuple[HeaderFindStep, ...] FIND_STEPSc.|D]}||}||cSy)z1Run find steps in order, returning the first hit.Nr")r1stepssteprMs rrun_find_stepsrbs*d  M rcltj|}|td|t|tS)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 rF RuntimeErrorrbr^)libnamer1s rlocate_nvidia_header_directoryrfs90  ! !' *D |.gZ011 $ ++rc8t|}|r |jSdS)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: 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)rfr)refounds rfind_nvidia_header_directoryris, +7 3E"5>>,,r)r%rrr)r+rr,rrbool)r1rr2rrr)r1rrLocatedHeaderDir | None)r1rr`r]rrk)rerrrk)rerrr)( __future__r functoolsrZr$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 rrHeaderFindSteprr-r8rArIrNrTr\r^r!rbcacherfrir"rrrxs# $! G BOK 11 1-.0@40GGH ; & , * & ,,:-r