Lines Matching defs:numWorkerThreads
105 AsyncParallelForPass(bool asyncDispatch, int32_t numWorkerThreads,
108 this->numWorkerThreads = numWorkerThreads;
118 MLIRContext *ctx, bool asyncDispatch, int32_t numWorkerThreads,
121 numWorkerThreads(numWorkerThreads),
129 int32_t numWorkerThreads;
807 if (numWorkerThreads >= 0)
808 numWorkerThreadsVal = b.create<arith::ConstantIndexOp>(numWorkerThreads);
816 // float overshardingFactor = numWorkerThreads <= 4 ? 8.0
817 // : numWorkerThreads <= 8 ? 4.0
818 // : numWorkerThreads <= 16 ? 2.0
819 // : numWorkerThreads <= 32 ? 1.0
820 // : numWorkerThreads <= 64 ? 0.8
930 patterns, asyncDispatch, numWorkerThreads,
943 int32_t numWorkerThreads,
945 return std::make_unique<AsyncParallelForPass>(asyncDispatch, numWorkerThreads,
950 RewritePatternSet &patterns, bool asyncDispatch, int32_t numWorkerThreads,
953 patterns.add<AsyncParallelForRewrite>(ctx, asyncDispatch, numWorkerThreads,