Home
last modified time | relevance | path

Searched defs:CUstream (Results 1 – 3 of 3) sorted by relevance

/llvm-project/mlir/lib/ExecutionEngine/
H A DCudaRuntimeWrappers.cpp240 mgpuMemAlloc(uint64_t sizeBytes, CUstream stream, bool isHostShared) { in mgpuMemAlloc() argument
249 mgpuMemFree(void * ptr,CUstream) mgpuMemFree() argument
560 mgpuCreateDnVec(intptr_t size,void * values,int32_t dtp,CUstream) mgpuCreateDnVec() argument
568 mgpuDestroyDnVec(void * v,CUstream) mgpuDestroyDnVec() argument
575 mgpuCreateDnMat(intptr_t rows,intptr_t cols,void * values,int32_t dtp,CUstream) mgpuCreateDnMat() argument
584 mgpuDestroyDnMat(void * m,CUstream) mgpuDestroyDnMat() argument
592 mgpuCreateCoo(intptr_t rows,intptr_t cols,intptr_t nnz,void * rowIdxs,void * colIdxs,void * values,int32_t itp,int32_t dtp,CUstream) mgpuCreateCoo() argument
605 mgpuCreateCooAoS(intptr_t rows,intptr_t cols,intptr_t nnz,void * idxs,void * values,int32_t itp,int32_t dtp,CUstream) mgpuCreateCooAoS() argument
618 mgpuCreateCsr(intptr_t rows,intptr_t cols,intptr_t nnz,void * rowPos,void * colIdxs,void * values,int32_t ptp,int32_t itp,int32_t dtp,CUstream) mgpuCreateCsr() argument
632 mgpuCreateCsc(intptr_t rows,intptr_t cols,intptr_t nnz,void * colPos,void * rowIdxs,void * values,int32_t ptp,int32_t itp,int32_t dtp,CUstream) mgpuCreateCsc() argument
646 mgpuCreateBsr(intptr_t brows,intptr_t bcols,intptr_t bnnz,intptr_t rBsz,intptr_t cBsz,void * rowPos,void * colIdxs,void * values,int32_t ptp,int32_t itp,int32_t dtp,CUstream) mgpuCreateBsr() argument
660 mgpuDestroySpMat(void * m,CUstream) mgpuDestroySpMat() argument
666 mgpuSpMVBufferSize(int32_t ma,void * a,void * x,void * y,int32_t ctp,CUstream) mgpuSpMVBufferSize() argument
684 mgpuSpMV(int32_t ma,void * a,void * x,void * y,int32_t ctp,void * buf,CUstream) mgpuSpMV() argument
699 mgpuSpMMBufferSize(int32_t ma,int32_t mb,void * a,void * b,void * c,int32_t ctp,CUstream) mgpuSpMMBufferSize() argument
718 mgpuSpMM(int32_t ma,int32_t mb,void * a,void * b,void * c,int32_t ctp,void * buf,CUstream) mgpuSpMM() argument
734 mgpuSDDMMBufferSize(int32_t ma,int32_t mb,void * a,void * b,void * c,int32_t ctp,CUstream) mgpuSDDMMBufferSize() argument
753 mgpuSDDMM(int32_t ma,int32_t mb,void * a,void * b,void * c,int32_t ctp,void * buf,CUstream) mgpuSDDMM() argument
768 mgpuSpGEMMCreateDescr(CUstream) mgpuSpGEMMCreateDescr() argument
775 mgpuSpGEMMDestroyDescr(void * s,CUstream) mgpuSpGEMMDestroyDescr() argument
782 mgpuSpGEMMWorkEstimation(void * s,int32_t ma,int32_t mb,void * a,void * b,void * c,int32_t ctp,intptr_t bs,void * buf,CUstream) mgpuSpGEMMWorkEstimation() argument
800 mgpuSpGEMMCompute(void * s,int32_t ma,int32_t mb,void * a,void * b,void * c,int32_t ctp,intptr_t bsz2,void * buf2,CUstream) mgpuSpGEMMCompute() argument
818 mgpuSpGEMMCopy(void * s,int32_t ma,int32_t mb,void * a,void * b,void * c,int32_t ctp,CUstream) mgpuSpGEMMCopy() argument
833 mgpuSpMatGetSize(void * m,void * r,void * c,void * n,CUstream) mgpuSpMatGetSize() argument
843 mgpuSetCsrPointers(void * m,void * p,void * c,void * v,CUstream) mgpuSetCsrPointers() argument
895 mgpuCreateCuSparseLtDnMat(void * dh,intptr_t rows,intptr_t cols,void * values,int32_t dtp,CUstream) mgpuCreateCuSparseLtDnMat() argument
908 mgpuDestroyCuSparseLtDnMat(void * dh,CUstream) mgpuDestroyCuSparseLtDnMat() argument
915 mgpuCusparseLtCreate2To4SpMat(void * sh,intptr_t rows,intptr_t cols,void * values,int32_t dtp,CUstream) mgpuCusparseLtCreate2To4SpMat() argument
928 mgpuDestroyCuSparseLtSpMat(void * sh,CUstream) mgpuDestroyCuSparseLtSpMat() argument
[all...]
/llvm-project/offload/plugins-nextgen/cuda/dynamic_cuda/
H A Dcuda.h33 typedef struct CUstream_st *CUstream; global() typedef
[all...]
/llvm-project/offload/plugins-nextgen/cuda/src/
H A Drtl.cpp73 CUresult cuMemAllocAsync(CUdeviceptr *ptr, size_t, CUstream) { *ptr = 0; } in cuMemAllocAsync() argument
[all...]