`iS dZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z mZmZmZddlZdadadadada daddddZejdddkZd ed dfd Zd Zd ZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*d ee ee ffdZ+ded ee ee ffdZ,ded e-fdZ.dZ/ded ee0e0e0ffd Z1d!eded"ed eej2j3fd#Z4ded"ed e5efd$Z6ded"ed e5efd%Z7d&Z8d'e0d(e0d eefd)Z9d*Z:d efd+Z;d efd,Z>> cBtdkrtatSNr) _cuda_path_get_cuda_pathrr get_cuda_pathr I R#%% rcBtdkrtatSr) _nvcc_path_get_nvcc_pathrrr get_nvcc_pathr%Qr!rcBtdkrtatSr) _rocm_path_get_rocm_pathrrr get_rocm_pathr)Yr!rcBtdkrtatSr) _hipcc_path_get_hipcc_pathrrrget_hipcc_pathr-as b%'' rcBtdkrtatSr) _cub_path _get_cub_pathrrr get_cub_pathr1isB!OO rcntjdd}tj|r|St jd}|s rr$r$s^ vt,,IIt <RW\\)U%C%C D D DDrcntjdd}tj|r|St jd}|zinclude/hipcubz) cupy_backends.cuda.apirLr6r9r=abspath__file__is_hipisdirrDr))rL current_dirr/rHs rr0r0s......'//"'//(";";< 7== [*HII K K "IIII"OO  RW]] Y(899&;&; !III rc*tjdrpt}|duo |ddk}t }|&|r|}n9t j|d}nd}|stj dtd |t jtddd }t j |r#td |n$td |d}d tjkr|td |t j|t j|d}t j|r&td|t j||8td |t j|dSdS|etd |t jdd}|t jz|zt jd<dSdSdS)Nwin32 packagingcondarBzZCUDA path could not be detected. Set CUDA_PATH environment variable if CuPy fails to load.z CUDA_PATH: {}cupy.datalibzWheel shared libraries: {}z%Not wheel distribution ({} not found))zAdding DLL search path: {}x64z&Adding DLL search path (for CUDA 13): zAdding to PATH: {}PATHr)rplatform startswithget_preload_configr r6r9rDwarningswarnrformatget_cupy_install_pathrQ version_infoadd_dll_directoryr:r7r8pathsep)configis_condar> cuda_bin_path wheel_libdircuda_bin_x64_pathr9s r_setup_win32_dll_directoryrms |w''0F#%%$&KF;,?7,J"OO   ? ) " Y > >  M & %&&& _ # #I . .///w|| ! # #VWe== 7== & & -44\BB C C C C 8??   L S% % %(188GGHHH$]333$&GLL$F$F!7>>"344<0-00111():;;;'188FFGGG$\22222(' ')00>>???z~~fb11%1BJ%>%E 6"""a0F0FZ('rctjtjtjt dS)Nz..)r6r9rNrDr=rOrrrrdrds; 7??  RW__X..55 7 77rctjdd}|tjdStj|S)a|Returns the directory where CUDA external libraries are installed. This environment variable only affects wheel installations. Shared libraries are looked up from `$CUPY_CUDA_LIB_PATH/$CUDA_VER/$LIB_NAME/$LIB_VER/{lib,lib64,bin}`, e.g., `~/.cupy/cuda_lib/11.2/cudnn/8.1.1/lib64/libcudnn.so.8.1.1`. The default $CUPY_CUDA_LIB_PATH is `~/.cupy/cuda_lib`. CUPY_CUDA_LIB_PATHNz~/.cupy/cuda_lib)r6r7r8r9 expanduserrN)cupy_cuda_lib_paths rget_cupy_cuda_lib_pathrssM(>+ & &t k  ay||sBBBrYct}| |ddkrtd|ddS|tvrtd|||vrtd|d dSt|td |dSd S) z(Returns if the preload can be attempted.NrUrVzNot preloading z( as this is not a pip wheel installationFzUnknown preload library: zPreload z not configured in wheelzPreload already attempted: T)r`r _preload_libsAssertionError)rYrhs r_can_attempt_preloadr"s ! !FF;/7:: Ls L L LMMMu ->>>??? & 5 5 5 5666uS% 03 0 0111u 4rc  tdtsdSit<t d}td|t td  d d}|D] tdd d d  fd d D}dkr( d}t | d|z}n-dkr' d}t| d|z}|D]}tj |s#td|D td|tj |t|<tdn#t$rM}t|j}d|d|d |d }t|t!j|Yd}~d}~wwxYwtd td d tj t <td#t$r }tdd|Yd}~d}~wwxYwdS)zPreload dependent shared libraries. The preload configuration file (cupy/.data/_wheel.json) will be added during the wheel build process. z"Preloading triggered for library: Ncudaz$CuPy wheel package built for CUDA {}zCuPy CUDA library directory: {}version filenames Looking for z version  ()c dg|],}tjd|-S)r)r6r9rD).0xrhrrfilenamerYrs r z$_preload_library..ZsM... GLL"F6NC!  ...r)rYlib64rBr min_pypi_versionr z"Rejected candidate (not found): {}zTrying to load Loadedz CuPy failed to preload library (z): zFile {} could not be foundz from default search pathzLibrary z could not be preloaded: )rrrr`rcrs_get_cutensor_from_wheel_get_nccl_from_wheelr6r9r:ctypesCDLL Exceptiontype__name__rarb) rY cuda_versionr libpath_candsrlibpathee_typerrhrrrrs ` @@@@r_preload_libraryr>s 3c 3 3444  $ $M#  ! !F&>L / 6 6| D DEEE/11 * 1 12D E EFFFSk)$Gs K(I4C4C @C @ @' @ @X @ @ @AAA ........- ... *  %c{+=> ()96&>JJ MF]]%c{+=> $%5vf~FF % C CG7>>'** 9@@IIJJJ #0w00111.4k'.B.B c"7+X # # #a)&w&&&&!"&&&S  c""""""""  # -44X>> ? ? ? F8FFF G G G C/5{8/D/D c"8,X C C CAAAaAABBBBBBBB C#A4C4Cs2 AG H%AH  H%1J J;J66J;rc tjd|d}t|d}t|dkrt|dnd}t|dkrt|dnd}|||fS)Nz[^\d]rZ)maxsplitr)resplitintlen)rpartsmajorminorpatchs r_parse_versionrsv HXw 3 3 3E aMME ZZ1__CaMMM!E ZZ1__CaMMM!E % rpackage_prefixrc :|d\}}|d|} tj|}n0#tjj$rt |d|dYdSwxYwt |j}t |}|d|dko5|d|dkp#|d|dko|d|dk} | s!t |d|d |d |jdSt |d|d |j|S) z Returns the distribution of the given package name and version installed via pip (e.g., cutensor-cuXX). If the package is not found or incompatible, returns None. .z-cuz wheel package (z) not installedNrrrz) incompatible: expected z, found z ) found: )r importlibmetadata distributionPackageNotFoundErrorrrr) rrrcuda_major_ver_pkgdistactualexpected is_compatibles r_find_compatible_wheelrs 3NA  0 0 0 0C!..s33   2  D D D D DEEEttDL ) )Fg&&Hq Xa[  1I # D AY(1+ % B&)x{*B   9999 99*.,99 : : :tN H HC H H$, H HIII KsA)A.-A.ctd||}|gStjdkr2|d|dd}n|d}t |gS)zx Returns the list of shared library path candidates for cuTENSOR installed via Pip (cutensor-cuXX package). r Nlinuxzcutensor/lib/libcutensor.so.rrzcutensor\bin\cutensor.dll)rrr^ locate_filerstr)rrr shared_libs rrrs "*gt < .s , , ,CHH , , ,r)rr^rrr)rrr shared_libss rrrs}  |w !-$ ? ?D |   A'--*<*=z,>  I 3E33!)<<    G / / /G / / /000!..x88   2 -...  |w$,"9"9W---"H"H&&'C'C'C'CDD !!##  ; < < <IK  !! ;DL ; ; ;<<< sE00%FFchd}dtjD}||z}dt|kr cupy-cuda80 cupy-cuda90 cupy-cuda91 cupy-cuda92 cupy-cuda100 cupy-cuda101 cupy-cuda102 cupy-cuda110 cupy-cuda111 cupy-cuda112 cupy-cuda113 cupy-cuda114 cupy-cuda115 cupy-cuda116 cupy-cuda117 cupy-cuda118 cupy-cuda11x cupy-cuda12x cupy-cuda13x cupy-rocm-4-0 cupy-rocm-4-1 cupy-rocm-4-2 cupy-rocm-4-3 cupy-rocm-5-0rWcDh|]}|jddS)NameN)rr8)rds r z1_detect_duplicate_installation..Os>DDDz~~fd33DDDrrz, z -------------------------------------------------------------------------------- CuPy may not function correctly because multiple CuPy packages are installed in your environment: a/ Follow these steps to resolve this issue: 1. For all packages listed above, run the following command to remove all existing CuPy installations: $ pip uninstall If you previously installed CuPy via conda, also run the following: $ conda uninstall cupy 2. Install the appropriate CuPy package. Refer to the Installation Guide for detailed instructions. https://docs.cupy.dev/en/stable/install.html -------------------------------------------------------------------------------- )rr distributionsrrDsortedrarb)knowninstalled_namescupy_installedcupy_packages_lists r_detect_duplicate_installationr,s   EBDD ) 2 @ @ B BDDDO_,N3~  !YYvn'='=>>        rcd}tjdkrG |tz }n4#t$r'}|dt |jd|z }Yd}~nd}~wwxYw|S)NaFailed to import CuPy. If you installed CuPy via wheels (cupy-cudaXXX or cupy-rocm-X-X), make sure that the package matches with the version of CUDA or ROCm installed. On Linux, you may need to set LD_LIBRARY_PATH environment variable depending on how you installed CUDA/ROCm. On Windows, try setting CUDA_PATH environment variable. Check the Installation Guide for details: https://docs.cupy.dev/en/latest/install.htmlrTz% The cause could not be identified: : )rr^_diagnose_win32_dll_loadrrr)rrs r_diagnose_import_errorrps 2C |w  +-- -CC    +77#++'(++ CCCCCC  Js& AAAc td}|dSddlm}tjd}|j|j|jg|j_|j|j_ dddtdg}tj d}|d D]} tj|}||j |t||d |d |ja#t $r|d |d Yt"$r;}|d |d t%|jd|dYd}~d}~wwxYwd|S)Nz _depends.jsonrr)wintypeskernel32z CUDA Path: zDLL dependencies:idependsz z -> z -> not foundz -> error (rrr)rvrrrHANDLELPWSTRDWORDGetModuleFileNameWargtypesrestyper create_unicode_buffer_handlerappendvalueFileNotFoundErrorrrrrD)rrrlinesfilepathrwdllrs rrrs_--Gr{:&&H(.,:H(*2.H'  'moo''  E +E22H "II I+d##C  ' ' Xs8}} M M M LL8d8888 9 9 9 9  3 3 3 LL1d111 2 2 2 2 2 I I I LLGdGGtAww/?GG1GGG H H H H H H H H I 99U  sAC//#E E1EE)=__doc__rimportlib.metadatarrzr6os.pathr^rr;rtypingrrrrrrarr#r'r+r/rurr7r8rrrr r%r)r-r1rr$r(r,r0rmrdrsr`rvboolrrrrr Distributionrlistrrrrrrrrrrrs   33333333333333     >  13 7 73 >cd $ E E E$FFF.3F3F3Fl777 ///"HT#s(^4$sCx.!9cd8JCJCJCZCE#sC-$8&)14 i -.Bcc"-#-S-T#Y----$@22C2DI2222jAAAH2#r