Home
last modified time | relevance | path

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

/llvm-project/mlir/lib/ExecutionEngine/
H A DCudaRuntimeWrappers.cpp704 cusparseDnMatDescr_t matB = reinterpret_cast<cusparseDnMatDescr_t>(b); mgpuSpMMBufferSize() local
723 cusparseDnMatDescr_t matB = reinterpret_cast<cusparseDnMatDescr_t>(b); mgpuSpMM() local
739 cusparseDnMatDescr_t matB = reinterpret_cast<cusparseDnMatDescr_t>(b); mgpuSDDMMBufferSize() local
758 cusparseDnMatDescr_t matB = reinterpret_cast<cusparseDnMatDescr_t>(b); mgpuSDDMM() local
787 cusparseSpMatDescr_t matB = reinterpret_cast<cusparseSpMatDescr_t>(b); mgpuSpGEMMWorkEstimation() local
805 cusparseSpMatDescr_t matB = reinterpret_cast<cusparseSpMatDescr_t>(b); mgpuSpGEMMCompute() local
823 cusparseSpMatDescr_t matB = reinterpret_cast<cusparseSpMatDescr_t>(b); mgpuSpGEMMCopy() local
946 auto matB = reinterpret_cast<cusparseLtDnMatHandleAndData *>(b); mgpuCuSparseLtSpMMBufferSize() local
1003 auto matB = reinterpret_cast<cusparseLtDnMatHandleAndData *>(b); mgpuCuSparseLtSpMM() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseGPUCodegen.cpp690 Value matB = genAllocCopy(rewriter, loc, bufB, tokens); in rewriteSpMM() local
965 Value matB = genAllocCopy(rewriter, loc, bufB, tokens); in rewrite2To4SpMM() local
1081 Value matB = genAllocCopy(rewriter, loc, bufB, tokens); rewriteSDDMM() local
[all...]
/llvm-project/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp369 SmallVector<Value> matB = matchAndRewrite() local
620 SmallVector<Value> matB = matchAndRewrite() local