o װie-@sddlZddlZddlZddlZddlZddlZddlZddlZejj j j Z ddZ Gddde ZGdddZdd d d Zd d ZddZdS)Nc Cs6z|WS|y}z t|WYd}~Sd}~wwN)repr)funcerrorserB/home/ubuntu/.local/lib/python3.10/site-packages/cupyx/_runtime.py_eval_or_errors r c@s$eZdZddZddZddZdS) _InstallInfocCs`|}|dur#tj|d}ttj|dttj|dd}nddd}||_||_dS)Nz.datalibinclude)r r )_get_cupy_package_rootospathjoin _dir_or_nonecupy_package_root data_paths)selfr data_rootrrrr__init__s z_InstallInfo.__init__cCs"||jvr td||j|S)NzInvalid data type: {})r ValueErrorformat)r data_typerrr get_data_path/s  z_InstallInfo.get_data_pathcCs0ztt}Wn tyYdSwtj|Sr)inspectgetfilecupy TypeErrorrrdirname)r cupy_pathrrrr 4s   z#_InstallInfo._get_cupy_package_rootN)__name__ __module__ __qualname__rrr rrrrr s r c@seZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZddddZddZdS) _RuntimeInfoNTfullcs<tj|_ts tj|_ntj|_tstj |_ ntj |_ t tjj |_tjj rWz ddl}|jj}WntyKddl}|j}Ynw|jd|d7_ttjjjtjjj|_ttjjjtjjj|_ttjjt|_d|_|rtddt|_zddlm tfddt|_!Wn tyYnwtd dt|_"td dt|_#d|_$|rtd dt|_$td dt|_%zddl&mm'}||_(Wn tyYnwtsztjj)*}Wn tyd}Ynw|durd |_+n t tjj,||_+tj-dr|rtj.dnd|_/|j/|_0zddl1mm2m3}||_/t|j*|j4|_0Wn tyDYnwtj-dr]|rVtj.dnd|_5|j5|_6z!ddl7mm2m8}||_5|9}|dkrzd}||_6Wn tyYnwtjj:|_;zddlWn tyYnwzddl?mm2m@} | 9|_AWn tyYnwzddlBmm2mC} | |_DWn tyYnwtjEjF|_Gz ddlH} | j|_IWn tyYnwtJjKjL|_Mz ddlN} | jKjL|_OWdStyYdSw)Nrz (CUDA Python: )z (available)cStjjtjjSr)rcudacublas getVersiondeviceget_cublas_handlerrrr z'_RuntimeInfo.__init__..cufftcsSr)r+rr0rrr.scS tjjSr)rr)curandr+rrrrr. cSr2r)rr)cusolver _getVersionrrrrr.r4cSr(r)rr)cusparser+r,get_cusparse_handlerrrrr.r/cSr2r)rr)nvrtcr+rrrrr.r4z(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_hipr) 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.cudar1 cufft_versioncurand_versioncusolver_versioncusparse_version nvrtc_versioncupy.cuda.thrustthrustthrust_versionr9r+cuda_extra_include_dirs$_get_include_dir_from_conda_or_wheel_can_attempt_preload_preload_librarycudnn_build_version cudnn_versioncupy_backends.cuda.libs.cudnnlibsr: CuDNNErrornccl_build_versionnccl_runtime_versioncupy_backends.cuda.libs.ncclr; 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&r) cuda_versionr_r]r:r;rkrqrtrwr|rrr0rrbs                    z_RuntimeInfo.__init__cCsdtfdtfd|jfdtsdndfd|jfd|jfd |jfd |jfd |jftr-d nd |j fd|j fd|j fd|j d|j dfd|jfg}|d|jfd|jfd|jfd|jfd|jfd|jfd|jfd|jfd|jfg 7}|d|jfd|jfd|jfd |jfd!|jfd"|jfg7}d#}ztjj !}Wn$tjj j"y}zd$|j#d#vrt$%d%|WYd}~nd}~wwt&|D]h}tj'|W}tjj (|}d&)||d'*f}d()||j+f}tr z|d)*} Wnt,yd*t-|d+} Ynwd,)|| f} nd-)||j.f} ||| |g7}Wdn 1s$wYqt/d.d/|Dd0} d1t-| d2} t01} |D]\} }| 2| )| |qC| 3S)3NOSzPython 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 CapabilitycSsg|]}t|dqS)r)len).0rrrr Psz(_RuntimeInfo.__str__..z{:z}: {} )4platformpython_versionr=r>rrr{r}r@rDrIrQrSrVrarWrYrZr[r\r]r`rorrrerfrjrkrurxrr)rNgetDeviceCountrPargswarningswarnrangeDevicegetDevicePropertiesrdecode pci_bus_idKeyErrorrFcompute_capabilitymaxioStringIOwritegetvalue)rrecords device_countr device_idr,propsrpci_busarchwidthfmtskvrrr__str__ s       z_RuntimeInfo.__str__)r!r"r#r=r@rIrQrSrVrWrYrZr[r\r]r`rarerfrjrkrorrrurxr{r}rrrrrrrrr$<s< *r$Tr%cCs t|dS)Nr%)r$r%rrrget_runtime_infoYs rcCstSr)r rrrrget_install_info]srcCstj|r|SdS)z$Returns None if path does not exist.N)rrisdir)rrrrras r)rrrrrr~r cupy_backendsr)apirNr>r objectr r$rrrrrrrs"  "