Lines Matching refs:waitOp
208 matchAndRewrite(gpu::WaitOp waitOp, ArrayRef<Value> operands,
222 matchAndRewrite(gpu::WaitOp waitOp, ArrayRef<Value> operands,
414 gpu::WaitOp waitOp, ArrayRef<Value> operands, in matchAndRewrite() argument
416 if (waitOp.asyncToken()) in matchAndRewrite()
417 return rewriter.notifyMatchFailure(waitOp, "Cannot convert async op."); in matchAndRewrite()
419 Location loc = waitOp.getLoc(); in matchAndRewrite()
426 rewriter.eraseOp(waitOp); in matchAndRewrite()
437 gpu::WaitOp waitOp, ArrayRef<Value> operands, in matchAndRewrite() argument
439 if (!waitOp.asyncToken()) in matchAndRewrite()
440 return rewriter.notifyMatchFailure(waitOp, "Can only convert async op."); in matchAndRewrite()
442 Location loc = waitOp.getLoc(); in matchAndRewrite()
446 for (auto pair : llvm::zip(waitOp.asyncDependencies(), operands)) { in matchAndRewrite()
453 rewriter.setInsertionPointToStart(waitOp->getBlock()); in matchAndRewrite()
466 rewriter.replaceOp(waitOp, {stream}); in matchAndRewrite()