Lines Matching refs:genericOp
37 interchangeGenericOpPrecondition(GenericOp genericOp, in interchangeGenericOpPrecondition() argument
41 genericOp.getNumLoops() != interchangeVector.size()) in interchangeGenericOpPrecondition()
45 genericOp.getContext()))) in interchangeGenericOpPrecondition()
51 mlir::linalg::interchangeGenericOp(RewriterBase &rewriter, GenericOp genericOp, in interchangeGenericOp() argument
53 if (failed(interchangeGenericOpPrecondition(genericOp, interchangeVector))) in interchangeGenericOp()
54 return rewriter.notifyMatchFailure(genericOp, "preconditions not met"); in interchangeGenericOp()
58 MLIRContext *context = genericOp.getContext(); in interchangeGenericOp()
64 rewriter.startRootUpdate(genericOp); in interchangeGenericOp()
66 llvm::make_scope_exit([&]() { rewriter.finalizeRootUpdate(genericOp); }); in interchangeGenericOp()
70 for (OpOperand *opOperand : genericOp.getInputAndOutputOperands()) { in interchangeGenericOp()
71 AffineMap m = genericOp.getTiedIndexingMap(opOperand); in interchangeGenericOp()
76 genericOp->setAttr(getIndexingMapsAttrName(), in interchangeGenericOp()
80 ArrayRef<Attribute> itTypes = genericOp.iterator_types().getValue(); in interchangeGenericOp()
86 genericOp->setAttr(getIteratorTypesAttrName(), in interchangeGenericOp()
90 if (genericOp.hasIndexSemantics()) { in interchangeGenericOp()
93 llvm::make_early_inc_range(genericOp.getBody()->getOps<IndexOp>())) { in interchangeGenericOp()
96 allIndices.reserve(genericOp.getNumLoops()); in interchangeGenericOp()
97 llvm::transform(llvm::seq<uint64_t>(0, genericOp.getNumLoops()), in interchangeGenericOp()
106 return genericOp; in interchangeGenericOp()