Lines Matching defs:asyncDispatch
70 // And a dispatch function depending on the `asyncDispatch` option.
105 AsyncParallelForPass(bool asyncDispatch, int32_t numWorkerThreads,
107 this->asyncDispatch = asyncDispatch;
118 MLIRContext *ctx, bool asyncDispatch, int32_t numWorkerThreads,
120 : OpRewritePattern(ctx), asyncDispatch(asyncDispatch),
128 bool asyncDispatch;
620 auto asyncDispatch = [&](OpBuilder &nestedBuilder, Location loc) {
643 b.create<scf::IfOp>(isSingleBlock, syncDispatch, asyncDispatch);
864 auto doDispatch = asyncDispatch ? doAsyncDispatch : doSequentialDispatch;
930 patterns, asyncDispatch, numWorkerThreads,
942 std::unique_ptr<Pass> mlir::createAsyncParallelForPass(bool asyncDispatch,
945 return std::make_unique<AsyncParallelForPass>(asyncDispatch, numWorkerThreads,
950 RewritePatternSet &patterns, bool asyncDispatch, int32_t numWorkerThreads,
953 patterns.add<AsyncParallelForRewrite>(ctx, asyncDispatch, numWorkerThreads,