Home
last modified time | relevance | path

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

/llvm-project/mlir/lib/ExecutionEngine/
H A DCudaRuntimeWrappers.cpp669 cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); mgpuSpMVBufferSize() local
687 cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); mgpuSpMV() local
703 cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); mgpuSpMMBufferSize() local
722 cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); mgpuSpMM() local
738 cusparseDnMatDescr_t matA = reinterpret_cast<cusparseDnMatDescr_t>(a); mgpuSDDMMBufferSize() local
757 cusparseDnMatDescr_t matA = reinterpret_cast<cusparseDnMatDescr_t>(a); mgpuSDDMM() local
786 cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); mgpuSpGEMMWorkEstimation() local
804 cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); mgpuSpGEMMCompute() local
822 cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); mgpuSpGEMMCopy() local
945 auto matA = reinterpret_cast<cusparseLtSpMatHandleAndData *>(a); mgpuCuSparseLtSpMMBufferSize() local
1002 auto matA = reinterpret_cast<cusparseLtSpMatHandleAndData *>(a); mgpuCuSparseLtSpMM() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseGPUCodegen.cpp963 Value matA = genAllocCopy(rewriter, loc, bufA, tokens); in rewrite2To4SpMM() local
1079 Value matA = genAllocCopy(rewriter, loc, bufA, tokens); in rewriteSDDMM() local
/llvm-project/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp367 SmallVector<Value> matA = matchAndRewrite() local
618 SmallVector<Value> matA = matchAndRewrite() local