`ie-ddlZddlZddlZddlZddlZddlZddlZddlZejj j j Z dZ Gdde ZGddZddd Zd Zd ZdS) NcZ |S#|$r}t|cYd}~Sd}~wwxYwN)repr)funcerrorses b/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/_runtime.py_eval_or_errorr sFtvv Awws *%**c eZdZdZdZdZdS) _InstallInfocP|}||tj|d}t tj|dt tj|dd}nddd}||_||_dS)Nz.datalibinclude)rr)_get_cupy_package_rootospathjoin _dir_or_nonecupy_package_root data_paths)selfr data_rootrs r __init__z_InstallInfo.__init__s 7799  ( %6@@I#BGLLE$B$BCC' Y (J(JKKJJ J "3$cr||jvr"td||j|S)NzInvalid data type: {})r ValueErrorformat)r data_types r get_data_pathz_InstallInfo.get_data_path/s9 DO + +4;;IFFGG Gy))rc tjt}n#t$rYdSwxYwtj|Sr)inspectgetfilecupy TypeErrorrrdirname)r cupy_paths r rz#_InstallInfo._get_cupy_package_root4sN --II   44 wy)))s  **N)__name__ __module__ __qualname__rrrrr r r sA%%%"*** *****rr ceZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdddZdZdS) _RuntimeInfoNTfullc tj|_ts$tj|_n#tj|_ts$tj |_ n#tj |_ ttjj |_tjj rA ddl}|jj}n#t&$rddl}|j}YnwxYw|xjd|dz c_t)tjjjtjjj|_t)tjjjtjjj|_t)tjjt8|_d|_|rt)dt8|_ ddlm t)fdt8|_!n#t&$rYnwxYwt)dt8|_"t)d t8|_#d|_$|rt)d t8|_$t)d t8|_% ddl&mcm'}||_(n#t&$rYnwxYwtsf tjj)*}n#t8$rd}YnwxYw|d |_+n%ttjj,||_+tj-d r5|r tj.d nd|_/|j/|_0 ddl1mcm2cm3}||_/t)|j*|j4|_0n#t&$rYnwxYwtj-dr5|r tj.dnd|_5|j5|_6 ddl7mcm2cm8}||_5|9}|dkrd}||_6n#t&$rYnwxYwtjj:|_; ddln#t&$rYnwxYw ddl?mcm2cm@} | 9|_An#t&$rYnwxYw ddlBmcm2cmC} | |_Dn#t&$rYnwxYwtjEjF|_G ddlH} | j|_In#t&$rYnwxYwtjKjL|_M ddlN} | jKjL|_OdS#t&$rYdSwxYw)Nrz (CUDA Python: )z (available)ctjjtjjSr)r#cudacublas getVersiondeviceget_cublas_handler*rr z'_RuntimeInfo.__init__..s/ (33I$6688::rcufftc,Sr)r4r8sr r7z'_RuntimeInfo.__init__..s((**rcHtjjSr)r#r2curandr4r*rr r7z'_RuntimeInfo.__init__..sDI$//11rcHtjjSr)r#r2cusolver _getVersionr*rr r7z'_RuntimeInfo.__init__..sDI&2244rctjjtjjSr)r#r2cusparser4r5get_cusparse_handler*rr r7z'_RuntimeInfo.__init__..s/ *55I$88::<<rcHtjjSr)r#r2nvrtcr4r*rr r7z'_RuntimeInfo.__init__..sDIO..00rz(NVRTC unavailable)cudnnz0(not loaded; try `import cupy.cuda.cudnn` first)ncclz/(not loaded; try `import cupy.cuda.nccl` first)z (unknown))Pr# __version__ cupy_versionis_hipr2 get_cuda_path cuda_path _environment get_rocm_path get_nvcc_path nvcc_pathget_hipcc_pathstrdriverget_build_versioncuda_build_version_is_cuda_python cuda.bindingsbindings ImportErrorr runtimedriverGetVersionCUDARuntimeErrorcuda_driver_versionruntimeGetVersioncuda_runtime_versionget_local_runtime_version Exceptioncuda_local_runtime_versioncublas_version cupy.cudar9 cufft_versioncurand_versioncusolver_versioncusparse_version nvrtc_versioncupy.cuda.thrustthrustthrust_versionrDr4cuda_extra_include_dirs$_get_include_dir_from_conda_or_wheel_can_attempt_preload_preload_librarycudnn_build_version cudnn_versioncupy_backends.cuda.libs.cudnnlibsrE CuDNNErrornccl_build_versionnccl_runtime_versioncupy_backends.cuda.libs.ncclrF get_versioncubcub_build_versioncupy.cuda.jitifyjitifyjitify_build_version cupy_backends.cuda.libs.cutensorcutensorcutensor_version"cupy_backends.cuda.libs.cusparselt cusparseltcusparselt_version_utilcython_build_vercython_build_versionCythoncython_versionnumpyversion full_version numpy_versionscipy scipy_version)rr.r2 cuda_versionrjrhrErFrvr|rrrrr9s @r rz_RuntimeInfo.__init__bs , ?!Y4466DNN!.<<>>DN @!.<<>>DNN!.==??DN#&di&6&H&H&J&J"K"K 9  + + - - I 0$$$$#}8  0 0 0 #/  0  # #'H'H'H'H H # ##1 I  . I  .$0$0 %3 I  / I  .%0%0!+9 I / ++' ,  "0::##D    ' ' ' ' ' '!/****I"7"7D      D - 1 1  !/ 4 4 !! !.  $2<<%%D ! , 0 0    - - - - - - - - -"(":":"<" >!227;;;;G(%)%="  9 9 9 9 9 9 9 9 9 9 9 9','>'>'@'@D $!/ %"2"4"4D      D    1 1& 9 9 D D!226::::F',0,C)  7 7 7 7 7 7 7 7 7 7 7 7&*&<&<&>&>D ##'#3#3#5#5 #q(('2$(> T  T %T55 UU%U,, U98U9V$$ V10V1 W"" W0/W0cdtjfdtjfd|jfdtsdndfd|jfd|jfd |jfd |jfd |jftrd nd |j fd|j fd|j fd|j d|j dfd|jfg}|d|jfd|jfd|jfd|jfd|jfd|jfd|jfd|jfd|jfg z }|d|jfd|jfd|jfd |jfd!|jfd"|jfgz }d#} t<jj !}nL#t<jj j"$r0}d$|j#d#vrtIj%d%|Yd}~nd}~wwxYwtM|D]0}t<j'|5}t<jj (|}d&)||d'*f}d()||j+f}tr\ |d)*} n(#tX$rd*t[|d+z} YnwxYwd,)|| f} nd-)||j.f} ||| |gz }dddn #1swxYwY2t_d.|Dd/z} d0t[| zd1z} taj1} |D].\} }| 2| )| |/| 3S)2NOSzPython Versionz CuPy Versionz CuPy Platformz NVIDIA CUDAzAMD ROCmz NumPy Versionz SciPy VersionzCython Build VersionzCython Runtime Versionz CUDA Rootz hipcc PATHz nvcc PATHzCUDA Build VersionzCUDA Driver VersionzCUDA Runtime Versionz (linked to CuPy) / z (locally installed)zCUDA Extra Include DirszcuBLAS Versionz cuFFT VersionzcuRAND VersionzcuSOLVER VersionzcuSPARSE Versionz NVRTC VersionzThrust VersionzCUB Build VersionzJitify Build VersionzcuDNN Build Versionz cuDNN VersionzNCCL Build VersionzNCCL Runtime VersionzcuTENSOR VersionzcuSPARSELt Build Versionr ErrorNoDevicez!Failed to detect number of GPUs: zDevice {} NamenamezDevice {} PCI Bus ID gcnArchNamegfxgcnArchzDevice {} ArchzDevice {} Compute Capabilityc8g|]}t|dS)r)len).0rs r z(_RuntimeInfo.__str__..Ps"0001S1YY000rz{:z}: {} )4platformpython_versionrHrIrrrrrKrOrTr\r^rarlrbrdrerfrgrhrkrzr}rprqrurvrrr#r2rYgetDeviceCountr[argswarningswarnrangeDevicegetDevicePropertiesrdecode pci_bus_idKeyErrorrQcompute_capabilitymaxioStringIOwritegetvalue)rrecords device_countr device_idr5propsrpci_busarchwidthfmtskvs r __str__z_RuntimeInfo.__str__ s9 H%'' ( x688 9 T. / 6Immz J d0 1 d0 1 #T%> ? %t': ; $. )# 4\\dn E !4#: ; "D$< = #,II2III '(D E' ,  t2 3 d0 1 t2 3 !6 7 !6 7 d0 1 t2 3 $"8 9 #T%> ?    "D$< = d0 1 !4#: ; #T%> ? !6 7 ')@ A     G9,;;==LLy 1 G G GafQi// E!EEFFF G|,, 1 1I!!),, 1 )==iHH(// ::f ,,..0188CC!,.7;$]3::<<#;;;$Sy)9%:%::;,33I>>EDD:AA)LL"57DD$00 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1"0000011A5SZZ)+ KMM & &DAq GGCJJq!$$ % % % %zz||sO(#E F%&FF A7K IK "J>K J?K  K K )r'r(r)rHrKrTr\r^rarbrdrerfrgrhrkrlrprqrurvrzr}rrrrrrrrr*rr r,r,<sLI !%NMNMN"MNMM#gggggRKKKKKrr,Tr-c"t|S)Nr-)r,r-s r get_runtime_inforYs T " " ""rctSr)r r*rr get_install_infor]s >>rcHtj|r|SdS)z$Returns None if path does not exist.N)rrisdir)rs r rras" w}}T 4r)r!rrrrrr# cupy_backendsr2apirYrIr objectr r,rrrr*rr rs      ' .*****6***DZZZZZZZZz"#####r