o װiW8 @sUddlZddlZddlmZdZdZdZdZee dfe d<d Z e d 7Z d Z d Z d ZeZddZddZddZddZddZedkredd ZeZWdn1sZwYdZeD]TZeeed ZeZWdn1s|wYdZeD]1Z e!Z"ej#$ee Z%e%Ze&Z'Wdn1swYee'ee"e eZdZqqce(ddZe(dd Zee)d!dZed"d#Ze*eWddS1swYdSdS)$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 cCsH||} |dkr |S||t|dkr |||d}n |Sq)NT()findlen)cu_hcu_funccu_sigr+V/home/ubuntu/.local/lib/python3.10/site-packages/cupyx/tools/_hipsparse_stub_mapper.pyget_idx_to_funcas r-cCs&|d}t|ddt|dS)N.rdr%)splitint) hip_versionr+r+r,get_hip_ver_numos r3cCsg}d}t|D]E\}}||krM|ds|dr ||q|d}|dkr.|d}|dkrB||||d|d}q||d|dq|S)N,)r#r%) enumerateendswithappendr&)r*cu_sig_processed skip_linelines break_idxr+r+r,merge_bad_broken_linesus    r>cCs4d|vr|}d|dddd|dd}d}npd|vr<|}d|dddd|dd}d}nRd|vrZ|}d|dddd|dd}d }n4d |vrx|}d|dddd|dd}d }nd |vr|}|d 7}|d|dddd|dddd7}d|dd}d}nd|vr|}|d7}|d|dddd|dddd7}d|dd}d}nd|vr|}d|dddd|dd}d}nd|vr|}|d7}||dddd7}d|dd}d}nyd|vr;|}|d7}|d|dddd7}d|dd}d}nSd|vra|}|d7}|d|dddd7}d |dd}d}n-d!|vr|d"kr|}d|dddd|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; 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 )r0)r<hip_sigdeclhip_funcargcastr+r+r,process_func_argssx$$$$..$     $rLc Cst|}g}t|D]I\}}|dkr8|s8|||dkr7|d|d|d|t|tq |drd}d} d } tD]>} | |vr| tvr| d r\d | d dn| } | |vr|}| | kru|}d | d| d}n|dkr{d}d} d } n nqEd} |dur| dkr|dkr|tn| r|dkr| r|d|n|d|| dkr|dks||| r|dkr| r|d|||d| dur| rt | q d|vr*| } z t | dksJWnt yt d| w| d}|d|d}d |d d}t||}t||}|dkr0|dkr0J|dkrE|dkrEt |dtjdd}n`|dkrZ|dkrZt |dtjdd}nK||dd}|dksjJ||||d}| d}|d d}|ddD]}|dt | d d|d7}q|dd}|| d<d }|d| ||d}d!|vr||d }d!|vsJ|r| d}t|}|dkr|d|d"|d}d}|D] }t||||\}}q|ddd}||||dkr|d|d#|d$q ||d|d!d%d&q d!|vrOd'|vr5q d(|vrId||dvrH||q ||q ||q d|dS))Ni1z#include z2#include // for HIP_VERSIONz2#include // for hipDataTypetypedefrAFTcuhipztypedef rF;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?r5 rreturnz return z( return HIPSPARSE_STATUS_NOT_SUPPORTED;z#endifz HIPSPARSE_STATUS_NOT_SUPPORTED;CUSPARSE_STATUSHIPSPARSE_STATUS_NOT_SUPPORTED)r3r6r8cudaDataType_converterdefault_return_code startswithrprocessed_typedefscusparseOrder_converteraddr0r'AssertionErrorprintr&r-sysstderrjoinr>rL)hip_hr(stubsr2init hip_stub_hir;old_line typedef_foundtypedef_neededthip_tsigr)rIr*rGcan_mapend_idx new_cu_sigr<rHr+r+r,mains                    $                 rr__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_HrTz"cupy_backends/hip/cupy_hipsparse.hw)+urllib.requesturllibratypingr cusparse_h cu_versions hipsparse_url hip_versionsstr__annotations__rrYr]rZsetr\r-r3r>rLrr__name__openfreadrerfcu_verformatr(xhip_ver splitlinesstubs_splittedrequesturlopenreqdecoderdreplacer&writer+r+r+r,sh      E#      "