Lines Matching defs:genericOp
36 interchangeGenericOpPrecondition(GenericOp genericOp,
40 genericOp.getNumLoops() != interchangeVector.size())
44 genericOp.getContext())))
50 mlir::linalg::interchangeGenericOp(RewriterBase &rewriter, GenericOp genericOp,
52 if (failed(interchangeGenericOpPrecondition(genericOp, interchangeVector)))
53 return rewriter.notifyMatchFailure(genericOp, "preconditions not met");
57 MLIRContext *context = genericOp.getContext();
63 rewriter.startOpModification(genericOp);
65 [&]() { rewriter.finalizeOpModification(genericOp); });
69 for (OpOperand &opOperand : genericOp->getOpOperands()) {
70 AffineMap m = genericOp.getMatchingIndexingMap(&opOperand);
75 genericOp.setIndexingMapsAttr(
79 ArrayRef<Attribute> itTypes = genericOp.getIteratorTypes().getValue();
84 genericOp.setIteratorTypesAttr(rewriter.getArrayAttr(itTypesVector));
87 if (genericOp.hasIndexSemantics()) {
90 llvm::make_early_inc_range(genericOp.getBody()->getOps<IndexOp>())) {
93 allIndices.reserve(genericOp.getNumLoops());
94 llvm::transform(llvm::seq<uint64_t>(0, genericOp.getNumLoops()),
103 return genericOp;