Lines Matching defs:waitOp

76     if (auto waitOp = llvm::dyn_cast<gpu::WaitOp>(op)) {
78         waitOp.addAsyncDependency(currentToken);
79       currentToken = waitOp.getAsyncToken();
197       if (auto waitOp = dyn_cast<gpu::WaitOp>(op)) {
198         if (!waitOp.getAsyncToken())
199           worklist.push_back(waitOp);
210       auto waitOp = worklist[i];
211       auto executeOp = waitOp->getParentOfType<async::ExecuteOp>();
214       SmallVector<Value, 4> dependencies = waitOp.getAsyncDependencies();
215       waitOp.erase();
285     auto waitOp = builder.create<gpu::WaitOp>(loc, Type{}, tokens);
287     // If the new waitOp is at the end of an async.execute region, add it to the
292       worklist.push_back(waitOp);