Home
last modified time | relevance | path

Searched refs:contractOp (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp288 LogicalResult matchAndRewrite(vector::ContractionOp contractOp, in matchAndRewrite()
296 if (llvm::size(contractOp.getMasks()) != 0) in matchAndRewrite()
304 auto oldIndexingMaps = contractOp.getIndexingMapsArray(); in matchAndRewrite()
307 auto oldIteratorTypes = contractOp.getIteratorTypes(); in matchAndRewrite()
323 SmallVector<Value> operands = {contractOp.getLhs(), contractOp.getRhs(), in matchAndRewrite()
324 contractOp.getAcc()}; in matchAndRewrite()
361 contractOp.getContext()); in matchAndRewrite()
363 contractOp.getLoc(), operands[it.index()], perm); in matchAndRewrite()
383 contractOp.getContext())); in matchAndRewrite()
395 rewriter.getArrayAttr(newIteratorTypes), contractOp.getKind()); in matchAndRewrite()
[all …]
H A DVectorUnrollDistribute.cpp313 LogicalResult matchAndRewrite(vector::ContractionOp contractOp, in matchAndRewrite()
315 auto targetShape = getTargetShape(options, contractOp); in matchAndRewrite()
318 auto dstVecType = contractOp.getResultType().cast<VectorType>(); in matchAndRewrite()
321 Location loc = contractOp.getLoc(); in matchAndRewrite()
323 AffineMap dstAffineMap = contractOp.getIndexingMapsArray()[accIndex]; in matchAndRewrite()
330 contractOp.getIteratorTypes().size(), contractOp, options); in matchAndRewrite()
336 SmallVector<Value, 4> slicesOperands(contractOp.getNumOperands()); in matchAndRewrite()
356 extractOperand(3, contractOp.getMasks()[0], lhsPermutationMap, in matchAndRewrite()
366 extractOperand(4, contractOp.getMasks()[1], rhsPermutationMap, in matchAndRewrite()
384 rewriter, loc, contractOp, slicesOperands, targetType); in matchAndRewrite()
[all …]
H A DVectorTransforms.cpp607 if (failed(filter(contractOp))) in matchAndRewrite()
668 Value newLhs = contractOp.getLhs(); in matchAndRewrite()
669 Value newRhs = contractOp.getRhs(); in matchAndRewrite()
670 Location loc = contractOp.getLoc(); in matchAndRewrite()
1039 Value lhs = contractOp.getLhs(); in matchAndRewrite()
1040 Value rhs = contractOp.getRhs(); in matchAndRewrite()
1052 contractOp.getContext()); in matchAndRewrite()
1060 contractOp, lhs, rhs, contractOp.getAcc(), in matchAndRewrite()
1096 Value lhs = contractOp.getLhs(); in matchAndRewrite()
1097 Value rhs = contractOp.getRhs(); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/test/lib/Dialect/Vector/
H A DTestVectorTransforms.cpp316 vector::ContractionOp contractOp = cast<vector::ContractionOp>(op); in runOnOperation() local
318 contractOp.getIteratorTypes().size(), 4); in runOnOperation()
319 Type lhsType = contractOp.getLhsType().getElementType(); in runOnOperation()
332 vector::ContractionOp contractOp = cast<vector::ContractionOp>(op); in runOnOperation() local
333 if (contractOp.getIteratorTypes().size() == unrollOrder.size()) in runOnOperation()
342 auto contractOp = dyn_cast<ContractionOp>(op); in runOnOperation() local
343 if (!contractOp) in runOnOperation()
345 return SmallVector<int64_t, 4>(contractOp.getIteratorTypes().size(), 2); in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp828 } else if (auto contractOp = dyn_cast<vector::ContractionOp>(op)) { in convertVectorToMMAOps() local
829 convertContractOp(contractOp, valueMapping); in convertVectorToMMAOps()