Home
last modified time | relevance | path

Searched refs:contractionOp (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp856 .Case([&](vector::ContractionOp contractionOp) { in convertVectorToNVVMCompatibleMMASync() argument
857 return convertContractOpToMmaSync(contractionOp, valueMapping); in convertVectorToNVVMCompatibleMMASync()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp915 vector::ContractionOp contractionOp = in matchAndRewrite() local
918 if (!contractionOp) in matchAndRewrite()
921 contractionOp.getAcc().getDefiningOp())) { in matchAndRewrite()
923 rewriter.getZeroAttr(contractionOp.getAcc().getType())) { in matchAndRewrite()
925 bvm.map(contractionOp.getAcc(), otherOperand); in matchAndRewrite()
927 cast<vector::ContractionOp>(rewriter.clone(*contractionOp, bvm)); in matchAndRewrite()