Lines Matching refs:waitOp
258 matchAndRewrite(gpu::WaitOp waitOp, OpAdaptor adaptor,
272 matchAndRewrite(gpu::WaitOp waitOp, OpAdaptor adaptor,
553 gpu::WaitOp waitOp, OpAdaptor adaptor, in matchAndRewrite() argument
555 if (waitOp.asyncToken()) in matchAndRewrite()
556 return rewriter.notifyMatchFailure(waitOp, "Cannot convert async op."); in matchAndRewrite()
558 Location loc = waitOp.getLoc(); in matchAndRewrite()
573 rewriter.eraseOp(waitOp); in matchAndRewrite()
583 gpu::WaitOp waitOp, OpAdaptor adaptor, in matchAndRewrite() argument
585 if (!waitOp.asyncToken()) in matchAndRewrite()
586 return rewriter.notifyMatchFailure(waitOp, "Can only convert async op."); in matchAndRewrite()
588 Location loc = waitOp.getLoc(); in matchAndRewrite()
593 llvm::zip(waitOp.asyncDependencies(), adaptor.getOperands())) { in matchAndRewrite()
616 rewriter.replaceOp(waitOp, {stream}); in matchAndRewrite()