Lines Matching refs:asyncDispatch
100 AsyncParallelForPass(bool asyncDispatch, int32_t numWorkerThreads, in AsyncParallelForPass()
102 this->asyncDispatch = asyncDispatch; in AsyncParallelForPass()
113 MLIRContext *ctx, bool asyncDispatch, int32_t numWorkerThreads, in AsyncParallelForRewrite()
115 : OpRewritePattern(ctx), asyncDispatch(asyncDispatch), in AsyncParallelForRewrite()
123 bool asyncDispatch; member
619 auto asyncDispatch = [&](OpBuilder &nestedBuilder, Location loc) { in doAsyncDispatch() local
643 b.create<scf::IfOp>(TypeRange(), isSingleBlock, syncDispatch, asyncDispatch); in doAsyncDispatch()
864 auto doDispatch = asyncDispatch ? doAsyncDispatch : doSequentialDispatch; in matchAndRewrite()
930 patterns, asyncDispatch, numWorkerThreads, in runOnOperation()
942 std::unique_ptr<Pass> mlir::createAsyncParallelForPass(bool asyncDispatch, in createAsyncParallelForPass() argument
945 return std::make_unique<AsyncParallelForPass>(asyncDispatch, numWorkerThreads, in createAsyncParallelForPass()
950 RewritePatternSet &patterns, bool asyncDispatch, int32_t numWorkerThreads, in populateAsyncParallelForPatterns() argument
953 patterns.add<AsyncParallelForRewrite>(ctx, asyncDispatch, numWorkerThreads, in populateAsyncParallelForPatterns()