Lines Matching defs:forallOp
1192 bool mayHaveZeroIterations(scf::ForallOp forallOp) {
1193 for (auto [lb, ub] : llvm::zip(forallOp.getMixedLowerBound(),
1194 forallOp.getMixedUpperBound())) {
1212 auto forallOp = cast<ForallOp>(op);
1217 if (mayHaveZeroIterations(forallOp))
1222 return state.isValueRead(forallOp.getTiedBlockArgument(&opOperand));
1233 auto forallOp = cast<ForallOp>(op);
1235 {{forallOp.getTiedOpResult(&opOperand), BufferRelation::Equivalent}}};
1246 auto forallOp = cast<ForallOp>(op);
1247 int64_t rank = forallOp.getRank();
1251 for (Value out : forallOp.getOutputs()) {
1259 rewriter.setInsertionPointToStart(forallOp.getBody());
1261 forallOp.getBody()->getArguments().drop_front(rank), buffers)) {
1265 forallOp.getLoc(), bbArg.getType(), buffer);
1271 rewriter.setInsertionPoint(forallOp);
1274 forallOp.getLoc(), forallOp.getMixedLowerBound(),
1275 forallOp.getMixedUpperBound(), forallOp.getMixedStep(),
1276 /*outputs=*/ValueRange(), forallOp.getMapping());
1287 replacementBbArgs.append(forallOp.getOutputs().size(), Value());
1288 rewriter.mergeBlocks(forallOp.getBody(), newForallOp.getBody(),
1300 auto forallOp = cast<ForallOp>(op);
1306 forallOp.getTiedOpOperand(bbArg)->get(), options, invocationStack);
1311 forallOp.getOutputs()[cast<OpResult>(value).getResultNumber()], options,
1316 auto forallOp = cast<ForallOp>(op);
1321 llvm::zip(forallOp.getMixedLowerBound(), forallOp.getMixedUpperBound(),
1322 forallOp.getMixedStep())) {