Home
last modified time | relevance | path

Searched defs:dtp (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/python/mlir/runtime/
H A Dnp_to_memref.py30 as_ctype(dtp) global() argument
/llvm-project/mlir/lib/ExecutionEngine/
H A DCudaRuntimeWrappers.cpp521 ALPHABETA(dtp,alpha,beta) global() argument
560 mgpuCreateDnVec(intptr_t size,void * values,int32_t dtp,CUstream) mgpuCreateDnVec() argument
574 mgpuCreateDnMat(intptr_t rows,intptr_t cols,void * values,int32_t dtp,CUstream) mgpuCreateDnMat() argument
591 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
895 mgpuCreateCuSparseLtDnMat(void * dh,intptr_t rows,intptr_t cols,void * values,int32_t dtp,CUstream) mgpuCreateCuSparseLtDnMat() argument
915 mgpuCusparseLtCreate2To4SpMat(void * sh,intptr_t rows,intptr_t cols,void * values,int32_t dtp,CUstream) mgpuCusparseLtCreate2To4SpMat() argument
[all...]
/llvm-project/mlir/lib/Conversion/GPUCommon/
H A DGPUToLLVMConversion.cpp1207 .create(loc, rewriter, {dims[0], dims[1], pTensor, dtp, stream}) in matchAndRewrite() local
1114 auto dtp = genConstInt32From(rewriter, loc, getCuSparseDataTypeFrom(dType)); matchAndRewrite() local
1233 auto dtp = genConstInt32From(rewriter, loc, getCuSparseDataTypeFrom(dType)); matchAndRewrite() local
1266 auto dtp = genConstInt32From(rewriter, loc, getCuSparseDataTypeFrom(dType)); matchAndRewrite() local
1289 auto dtp = genConstInt32From(rewriter, loc, getCuSparseDataTypeFrom(dType)); matchAndRewrite() local
1679 auto dtp = genConstInt32From(rewriter, loc, getCuSparseDataTypeFrom(dType)); matchAndRewrite() local
1712 auto dtp = genConstInt32From(rewriter, loc, getCuSparseDataTypeFrom(dType)); matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Utils/
H A DMerger.cpp1235 Type dtp = exp(e).val.getType(); in inferType() local