Searched defs:modeA (Results 1 – 2 of 2) sorted by relevance
/llvm-project/mlir/lib/ExecutionEngine/ |
H A D | CudaRuntimeWrappers.cpp | 668 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 D | GPUToLLVMConversion.cpp | 1452 {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...] |