o ºiW8ã @sØUddlZddlZddlmZdZdZdZdZee dfe d<d Z e d 7Z d Z d Z d ZeƒZdd„Zdd„Zdd„Zdd„Zdd„Zedkrêeddƒ Ze ¡ZWdƒn1sZwYdZeD]TZee e¡dƒ Ze ¡ZWdƒn1s|wYdZeD]1Z e !¡Z"ej# $e e ¡¡Z%e%Ze ¡ &¡Z'Wdƒn1s§wYee'ee"e eƒZdZq…qce (dd¡Ze (dd ¡Zee )d!¡d…Zed"d#ƒZe *e¡WdƒdS1sãwYdSdS)$éN)ÚTuplez&/usr/local/cuda-{0}/include/cusparse.h)z11.3z11.0z10.2zehttps://raw.githubusercontent.com/ROCmSoftwarePlatform/hipSPARSE/rocm-{0}/library/include/hipsparse.h)z3.5.0z3.7.0z3.8.0z3.9.0z4.0.0z4.2.0.Útypedefs)ÚcusparseIndexBase_tÚcusparseStatus_tÚcusparseHandle_tÚcusparseMatDescr_tÚ csrsv2Info_tÚ csrsm2Info_tÚ csric02Info_tÚ bsric02Info_tÚcsrilu02Info_tÚbsrilu02Info_tÚcsrgemm2Info_tÚcusparseMatrixType_tÚcusparseFillMode_tÚcusparseDiagType_tÚcusparsePointerMode_tÚcusparseAction_tÚcusparseDirection_tÚcusparseSolvePolicy_tÚcusparseOperation_t) ÚcusparseSpVecDescr_tÚcusparseDnVecDescr_tÚcusparseSpMatDescr_tÚcusparseDnMatDescr_tÚcusparseIndexType_tÚcusparseFormat_tÚcusparseOrder_tÚcusparseSpMVAlg_tÚcusparseSpMMAlg_tÚcusparseSparseToDenseAlg_tÚcusparseDenseToSparseAlg_tÚcusparseCsr2CscAlg_taú #if HIP_VERSION >= 402 static hipDataType convert_hipDatatype(cudaDataType type) { switch(static_cast(type)) { case 2 /* CUDA_R_16F */: return HIP_R_16F; case 0 /* CUDA_R_32F */: return HIP_R_32F; case 1 /* CUDA_R_64F */: return HIP_R_64F; case 6 /* CUDA_C_16F */: return HIP_C_16F; case 4 /* CUDA_C_32F */: return HIP_C_32F; case 5 /* CUDA_C_64F */: return HIP_C_64F; default: throw std::runtime_error("unrecognized type"); } } #endif a| #if HIP_VERSION >= 402 typedef enum {} cusparseOrder_t; static hipsparseOrder_t convert_hipsparseOrder_t(cusparseOrder_t type) { switch(static_cast(type)) { case 1 /* CUSPARSE_ORDER_COL */: return HIPSPARSE_ORDER_COLUMN; case 2 /* CUSPARSE_ORDER_ROW */: return HIPSPARSE_ORDER_ROW; default: throw std::runtime_error("unrecognized type"); } } z[ #if HIP_VERSION < 401 #define HIPSPARSE_STATUS_NOT_SUPPORTED (hipsparseStatus_t)10 #endif cCsH| |¡} |dkr |S||t|ƒdkr | ||d¡}n |Sq)NTéÿÿÿÿú(é)ÚfindÚlen)Úcu_hÚcu_funcÚcu_sig©r+ú_/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupyx/tools/_hipsparse_stub_mapper.pyÚget_idx_to_funcas üùr-cCs&| d¡}t|dƒdt|dƒS)NÚ.rédr%)ÚsplitÚint)Ú hip_versionr+r+r,Úget_hip_ver_numos r3cCs g}d}t|ƒD]E\}}||krM| d¡s| d¡r | |¡q| d¡}|dkr.| d¡}|dkrB| |||d¡|d}q| |d|d…¡q|S)Nú,ú)r#r%)Ú enumerateÚendswithÚappendr&)r*Úcu_sig_processedÚ skip_lineÚlineÚsÚ break_idxr+r+r,Úmerge_bad_broken_linesus    €r>cCs4d|vr| ¡}d|ddd…d|dd}d}npd|vr<| ¡}d|ddd…d|dd}d}nRd|vrZ| ¡}d|ddd…d|dd}d }n4d |vrx| ¡}d|ddd…d|dd}d }nd |vr¦| ¡}|d 7}|d|ddd…›d|ddd…›d7}d|dd}d}nèd|vrÔ| ¡}|d7}|d|ddd…›d|ddd…›d7}d|dd}d}nºd|vrñ| ¡}d|ddd…d|dd}d}nd|vr| ¡}|d7}||ddd…d7}d|dd}d}nyd|vr;| ¡}|d7}|d|ddd…d7}d|dd}d}nSd|vra| ¡}|d7}|d|ddd…d7}d |dd}d}n-d!|vr„|d"kr„| ¡}d|ddd…d|dd}d#}n | ¡}|d}d}|||d$7}||fS)%Nzconst cuComplex*r$r#r5z#reinterpret_castzconst cuDoubleComplex*z)reinterpret_castz cuComplex*zreinterpret_castzcuDoubleComplex*z#reinterpret_castÚ cuComplexz hipComplex blah; z blah.x=z .x; blah.y=z.y; ÚblahÚÚcuDoubleComplexz hipDoubleComplex blah; z cudaDataType*zreinterpret_castÚ cudaDataTypez) hipDataType blah = convert_hipDatatype(z); zcusparseOrder_t*z hipsparseOrder_t blah2 = zconvert_hipsparseOrder_t(*z&blah2rzconvert_hipsparseOrder_t(Úblah2z const void*ÚhipsparseSpVV_bufferSizezconst_castú )r0)r<Úhip_sigÚdeclÚhip_funcÚargÚcastr+r+r,Úprocess_func_args‹sx$$$$..$     $rLc Cs¸t|ƒ}g}t|ƒD]I\}}|dkr8|s8| |¡|dkr7| d¡| d¡| d¡| t¡| t¡q | d¡rád}d} d } tD]>} | |vrƒ| tvrƒ|  d ¡r\d | d d…n| } | |vr€|}| | kru|}d | d| d}n|dkr{d}d} d } n nqEd} |durÕ| dkr˜|dkr˜| t¡n| r±|dkr±| r©| d|›¡n| d|›¡| dkr¹|dks¾| |¡| rÕ|dkrÕ| rÐ| d¡| |¡| d¡| durà| ràt  | ¡q d|vr*|  ¡} z t | ƒdksóJ‚Wnt yt d| ›ƒ‚w| d}|d| d¡…}d |d d…}t||ƒ}t||ƒ}|dkr0|dkr0J‚|dkrE|dkrEt |dtjdd}n`|dkrZ|dkrZt |dtjdd}nK||d… d¡}|dksjJ‚||||d…}|  d¡}|d d}|dd…D]}|dt | d ƒd|d7}q…|dd…}|| d<d }| d | ¡¡||d}d!|vrÅ||d }d!|vsÅJ‚|r|  d¡}t|ƒ}|dkrÞ| d|›¡d"|d}d}|D] }t||||ƒ\}}qè|dd…d}| ||¡|dkr| d¡| d#¡| d$¡q | |d| d!¡d%…d&¡q d!|vrOd'|vr5q d(|vrId||dvrH| |¡q | |¡q | |¡q d |¡dS))Néi1z#include z2#include // for HIP_VERSIONz2#include // for hipDataTypeÚtypedefrAFTÚcuÚhipéztypedef rFú;ri’z#if HIP_VERSION >= z#if HIP_VERSION < z#elsez#endif z...zsig is r%r$r#z%not found in cuSPARSE, maybe removed?)Úfilez,not found in hipSPARSE, maybe not supported?r5Ú rÚreturnz return z( return HIPSPARSE_STATUS_NOT_SUPPORTED;z#endiféz HIPSPARSE_STATUS_NOT_SUPPORTED;ÚCUSPARSE_STATUSÚHIPSPARSE_STATUS_NOT_SUPPORTED)r3r6r8ÚcudaDataType_converterÚdefault_return_codeÚ startswithrÚprocessed_typedefsÚcusparseOrder_converterÚaddr0r'ÚAssertionErrorÚprintr&r-ÚsysÚstderrÚjoinr>rL)Úhip_hr(Ústubsr2ÚinitÚ hip_stub_hÚir;Úold_lineÚ typedef_foundÚtypedef_neededÚtÚhip_tÚsigr)rIr*rGÚcan_mapÚend_idxÚ new_cu_sigr<rHr+r+r,ÚmainÐs   ÿÿ  € €         € þ  ÿÿ  $        ÿ  ÿ €ÿÿ    €  rrÚ__main__z"cupy_backends/stub/cupy_cusparse.hÚrFTz#define CUSPARSE_VERSION -1zg#define CUSPARSE_VERSION (hipsparseVersionMajor*100000+hipsparseVersionMinor*100+hipsparseVersionPatch)Ú"INCLUDE_GUARD_STUB_CUPY_CUSPARSE_HÚ"INCLUDE_GUARD_HIP_CUPY_HIPSPARSE_HrTz"cupy_backends/hip/cupy_hipsparse.hÚw)+Úurllib.requestÚurllibraÚtypingrÚ cusparse_hÚ cu_versionsÚ hipsparse_urlÚ hip_versionsÚstrÚ__annotations__rrYr]rZÚsetr\r-r3r>rLrrÚ__name__ÚopenÚfÚreadrerfÚcu_verÚformatr(ÚxÚhip_verÚ splitlinesÚstubs_splittedÚrequestÚurlopenÚreqÚdecoderdÚreplacer&Úwriter+r+r+r,Úsh      E#  ÿ ÿÿø þþ  "ÿá