`iW8 UddlZddlZddlmZdZdZdZdZee dfe d<d Z e d z Z d Z d Z d ZeZdZdZdZdZdZedkredd5ZeZdddn #1swxYwYdZeD]Zeeed5ZeZdddn #1swxYwYdZeD]Z e!Z"ej#$ee Z%e%5Ze&Z'dddn #1swxYwYee'ee"e eZdZe(ddZe(ddZee)ddZedd5Ze*eddddS#1swxYwYdSdS)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_tcusparseStatus_tcusparseHandle_tcusparseMatDescr_t csrsv2Info_t csrsm2Info_t csric02Info_t bsric02Info_tcsrilu02Info_tbsrilu02Info_tcsrgemm2Info_tcusparseMatrixType_tcusparseFillMode_tcusparseDiagType_tcusparsePointerMode_tcusparseAction_tcusparseDirection_tcusparseSolvePolicy_tcusparseOperation_t) cusparseSpVecDescr_tcusparseDnVecDescr_tcusparseSpMatDescr_tcusparseDnMatDescr_tcusparseIndexType_tcusparseFormat_tcusparseOrder_tcusparseSpMVAlg_tcusparseSpMMAlg_tcusparseSparseToDenseAlg_tcusparseDenseToSparseAlg_tcusparseCsr2CscAlg_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 c||} |dkrn8||t|zdkr|||dz}nn?|S)NT()findlen)cu_hcu_funccu_sigs v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/tools/_hipsparse_stub_mapper.pyget_idx_to_funcr.asg YYw  F R<<  &W% &# - -YYwq11FF   Mc|d}t|ddzt|dzS)N.rdr')splitint) hip_versions r-get_hip_ver_numr6os=##C((K {1~   $s;q>':': ::r/cg}d}t|D]\}}||kr|ds|dr||K|d}|dkr|d}|dkr'||||dzz|dz}||d|dz|S)N,)r%r') enumerateendswithappendr()r,cu_sig_processed skip_linelines break_idxs r-merge_bad_broken_linesrBusIV$$ = =a 9  zz# =!**S// = ''****FF3KK ?? !s I??$++AtAv,>??? $QII%++Aly{lO<<< r/cd|vr=|}d|dddzdz|ddz}d}nd|vr=|}d|dddzdz|ddz}d}nd|vr=|}d|dddzdz|ddz}d }nud |vr=|}d|dddzdz|ddz}d }n4d |vrV|}|d z }|d|dddd|ddddz }d|ddz}d}nd|vrV|}|dz }|d|dddd|ddddz }d|ddz}d}nd|vr=|}d|dddzdz|ddz}d}n?d|vrC|}|dz }||ddddzz }d|ddz}d}nd|vrF|}|dz }|d|dddzdzz }d|ddz}d}nd|vrF|}|dz }|d|dddzdzz }d |ddz}d}ndd!|vrB|d"kr<|}d|dddzdz|ddz}d#}n|}|d}d}|||zd$zz }||fS)%Nzconst cuComplex*r&r%r9z#reinterpret_castzconst cuDoubleComplex*z)reinterpret_castz cuComplex*zreinterpret_castzcuDoubleComplex*z#reinterpret_cast cuComplexz hipComplex blah; z blah.x=z .x; blah.y=z.y; blahcuDoubleComplexz 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 )r3)r@hip_sigdeclhip_funcargcasts r-process_func_argsrQs Q GGIIAbE#2#J$quRy04 !Q & & GGIIAbE#2#J$quRy0:    GGIIAbE#2#J$quRy0. q GGIIAbE#2#J$quRy04    GGII && GAbE#2#JGGaeCRCjGGGGquRy  a   GGII ,, GAbE#2#JGGaeCRCjGGGGquRy  A   GGIIAbE#2#J$quRy0/ 1   GGII ;; "crc V##quRy  q GGII -- ,quSbSz9FBB2r" a   GGII -- +aeCRCj86AA"b ! 1  666 GGIIAbE#2#J$quRy0" GGIIe s S !G D=r/c8 t|}g}t|D]a\}}|dkr|s|||dkrs|d|d|d|t|t|dr^d}d} d } t D]`} | |vrZ| tvrQ| d r d | d dzn| } | |vr&|}| | kr|}d | zdz| zdz}n |dkrd}d} d } n nad} || dkr!|dkr|tn;| r9|dkr3| r|d|n|d|| dkr|dks||| rG|dkrA| r*|d|||d| | rt | d|vr| } t| dksJn!#t$rtd| wxYw| d}|d|d}d |d dz}t||}t||}|dkr|dkrJ|dkr%|dkrt|dt jd}n|dkr%|dkrt|dt jd}n||dd}|dksJ||||zdz}| d}|d dz}|ddD]&}|dt| d dzz|zdzz }'|dd}|| d<d }|d| ||dz}d!|vr||d z}d!|vsJ|r| d}t'|}|dkr|d|d"|zdz}d}|D]}t)||||\}}|dddz}|||z|dkr?|d|d#|d$||d|d!d%zd&zd!|vrEd'|vr d(|vr$d||dz vr||5||L||cd|dzS))Ni1z#include z2#include // for HIP_VERSIONz2#include // for hipDataTypetypedefrFFTcuhipztypedef rK;riz#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?r9 rreturnz return z( return HIPSPARSE_STATUS_NOT_SUPPORTED;z#endifz HIPSPARSE_STATUS_NOT_SUPPORTED;CUSPARSE_STATUSHIPSPARSE_STATUS_NOT_SUPPORTED)r6r:r<cudaDataType_converterdefault_return_code startswithrprocessed_typedefscusparseOrder_converteraddr3r)AssertionErrorprintr(r.sysstderrjoinrBrQ)hip_hr*stubsr5init hip_stub_hir?old_line typedef_foundtypedef_neededthip_tsigr+rNr,rLcan_mapend_idx new_cu_sigr@rMs r-mainrxs!+..KJU##W$W$4 66$6   d # # #c!!!!":;;;!!HJJJ!!HJJJ!!"8999!!"5666 __Y ' 'J $H!M!N  99*>>>"N{S'8'8%N"))*M *M*MNNNN"))*L{*L*LMMM...;#3E3E%%d+++ 2[3%6%6%4"))'222"))(333%%j111}}"&&q))) d]]**,,C 3xx1}}}}}!   ooo&&&  !fG0w||C0001Gwqrr{*H%T733F%eX66G||2 u2'R--gF:''''2'R--h N:''''vww-,,S11"}}}}fVG^A%556 d++#AY- GGA3#c!f++/#:a"?$"FFJJ#CRCA   chhsmm , , ,1:Dt##QqSz4'''' <d++/77#%%%%&IK&I&IJJJ%03644A$57D(%4%4MGTT!#2#,,!!$w,///#%%%%g...%%BDDD%%h///!!0499X..q0019:<<<<   D((1T99eAaCj((%%d+++!!$''''   d # # # # IIj ! !T ))s I""J__main__z"cupy_backends/stub/cupy_cusparse.hrFTz#define CUSPARSE_VERSION -1zg#define CUSPARSE_VERSION (hipsparseVersionMajor*100000+hipsparseVersionMinor*100+hipsparseVersionPatch)"INCLUDE_GUARD_STUB_CUPY_CUSPARSE_H"INCLUDE_GUARD_HIP_CUPY_HIPSPARSE_HrZz"cupy_backends/hip/cupy_hipsparse.hw)+urllib.requesturllibrgtypingr cusparse_h cu_versions hipsparse_url hip_versionsstr__annotations__rr_rcr`setrbr.r6rBrQrx__name__openfreadrkrlcu_verformatr*xhip_ver splitlinesstubs_splittedrequesturlopenreqdecoderjreplacer(writer/r-rs 06 & B E  S/ <  && SUU   ;;; ,BBBJ_*_*_*D z 2C 8 8A D   T*##F++S 1 1 Q6688D                #  G"--//N.(()=)=g)F)FGGC *)) * * * * * * * * * * * * * * *DngtDDEDD  MM% # % %E MM,, . .E %**T""## $E 2C 8 8A ?sHA??BB/CC C $'EE E G--G14G1