Home
last modified time | relevance | path

Searched defs:modeA (Results 1 – 2 of 2) sorted by relevance

/llvm-project/mlir/lib/ExecutionEngine/
H A DCudaRuntimeWrappers.cpp668 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSpMVBufferSize() local
686 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSpMV() local
701 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSpMMBufferSize() local
720 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSpMM() local
736 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSDDMMBufferSize() local
755 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSDDMM() local
784 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSpGEMMWorkEstimation() local
802 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSpGEMMCompute() local
820 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuSpGEMMCopy() local
953 cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); mgpuCuSparseLtSpMMBufferSize() local
[all...]
/llvm-project/mlir/lib/Conversion/GPUCommon/
H A DGPUToLLVMConversion.cpp1452 {bufferSize, modeA, modeB, adaptor.getSpmatA(), in matchAndRewrite() local
1333 auto modeA = genConstInt32From(rewriter, loc, op.getModeA()); matchAndRewrite() local
1353 auto modeA = genConstInt32From(rewriter, loc, adaptor.getModeA()); matchAndRewrite() local
1373 auto modeA = genConstInt32From(rewriter, loc, adaptor.getModeA()); matchAndRewrite() local
1430 auto modeA = genConstInt32From(rewriter, loc, adaptor.getModeA()); matchAndRewrite() local
1498 auto modeA = genConstInt32From(rewriter, loc, adaptor.getModeA()); matchAndRewrite() local
1551 auto modeA = genConstInt32From(rewriter, loc, adaptor.getModeA()); matchAndRewrite() local
1590 auto modeA = genConstInt32From(rewriter, loc, adaptor.getModeA()); matchAndRewrite() local
[all...]