[flang-commits] [flang] [flang][cuda] Add shape to cuf.data_transfer operation (PR #104631)

Slava Zakharin via flang-commits flang-commits at lists.llvm.org
Fri Aug 16 12:22:16 PDT 2024


================
@@ -99,6 +99,11 @@ llvm::LogicalResult cuf::AllocateOp::verify() {
 llvm::LogicalResult cuf::DataTransferOp::verify() {
   mlir::Type srcTy = getSrc().getType();
   mlir::Type dstTy = getDst().getType();
+  if (!getShape().empty()) {
+    if (!fir::isa_ref_type(srcTy) || fir::isa_ref_type(dstTy))
----------------
vzakhari wrote:

?

```suggestion
    if (!fir::isa_ref_type(srcTy) || !fir::isa_ref_type(dstTy))
```

https://github.com/llvm/llvm-project/pull/104631


More information about the flang-commits mailing list