Lines Matching defs:allocOp

364   matchAndRewrite(gpu::AllocOp allocOp, OpAdaptor adaptor,
756     gpu::AllocOp allocOp, OpAdaptor adaptor,
759   MemRefType memRefType = allocOp.getType();
761   if (failed(areAllLLVMTypes(allocOp, adaptor.getOperands(), rewriter)) ||
765   auto loc = allocOp.getLoc();
767   bool isShared = allocOp.getHostShared();
769   if (isShared && allocOp.getAsyncToken())
771         allocOp, "Host Shared allocation cannot be done async");
772   if (!isShared && failed(isAsyncWithOneDependency(rewriter, allocOp)))
804   if (allocOp.getAsyncToken()) {
806     rewriter.replaceOp(allocOp, {memRefDescriptor, stream});
808     rewriter.replaceOp(allocOp, {memRefDescriptor});