Lines Matching refs:waitOp
69 if (auto waitOp = llvm::dyn_cast<gpu::WaitOp>(op)) { in visit() local
71 waitOp.addAsyncDependency(currentToken); in visit()
72 currentToken = waitOp.asyncToken(); in visit()
188 if (auto waitOp = dyn_cast<gpu::WaitOp>(op)) { in operator ()() local
189 if (!waitOp.asyncToken()) in operator ()()
190 worklist.push_back(waitOp); in operator ()()
201 auto waitOp = worklist[i]; in ~DeferWaitCallback() local
202 auto executeOp = waitOp->getParentOfType<async::ExecuteOp>(); in ~DeferWaitCallback()
205 SmallVector<Value, 4> dependencies = waitOp.asyncDependencies(); in ~DeferWaitCallback()
206 waitOp.erase(); in ~DeferWaitCallback()
277 auto waitOp = builder.create<gpu::WaitOp>(loc, Type{}, tokens); in addAsyncDependencyAfter() local
284 worklist.push_back(waitOp); in addAsyncDependencyAfter()