
    bi                     v    S SK r S SKrS SKrS SKJr  S SKJr  S SKJr  \ R                  S\
S\\
   4S j5       rg)    N)Optional)
IS_WINDOWS)find_sub_dirs_all_sitepackageslibnamereturnc                 D   U S:w  a  [        SU < 35      eU S:X  a  [        (       a  g Sn[        U5       HK  n[        R                  R                  US5      n[        R                  R                  U5      (       d  MI  Us  $    [        R                  R                  S5      nU(       a  [        R                  R                  U5      (       af  [        R                  R                  US5      n[        R                  R                  US5      n[        R                  R                  U5      (       a  U$ [        [        R                  " S5      SS	9 HK  n[        R                  R                  US5      n[        R                  R                  U5      (       d  MI  Us  $    g )
NnvshmemzUNKNOWN libname=)nvidiar	   includez	nvshmem.hCONDA_PREFIXr   z/usr/include/nvshmem_*T)reverse)RuntimeErrorr   r   ospathjoinisfileenvirongetisdirsortedglob)r   nvidia_sub_dirshdr_dirnvshmem_h_pathconda_prefixs        f/home/james-whalen/.local/lib/python3.13/site-packages/cuda/pathfinder/_headers/find_nvidia_headers.pyfind_nvidia_header_directoryr      s%   ).gZ011)

 7O1/Bg{;77>>.))N C
 ::>>.1Ll33'',,|Y7g{;77>>.))N$))$<=tLg{;77>>.))N M
     )	functoolsr   r   typingr   3cuda.pathfinder._dynamic_libs.supported_nvidia_libsr   $cuda.pathfinder._utils.find_sub_dirsr   cachestrr    r   r   <module>r&      sC      	  J O # (3-  r   