[Mlir-commits] [mlir] Fix bug in gpu.memcpy lowering for dynamically shaped operands. (PR #128820)

[Arnab Dutta]

================
@@ -17,3 +17,23 @@ module attributes {gpu.container_module} {
     return
   }
 }
+
+// -----
+
+module attributes {gpu.container_module} {
+
+  // CHECK: func @dynamic
+  func.func @dynamic(%dst : memref<?x?xf32, 1>, %src : memref<?x?xf32>) {
+    // CHECK: %[[T0:.*]] = llvm.call @mgpuStreamCreate
+    %t0 = gpu.wait async
+    %t1 = gpu.memcpy async [%t0] %dst, %src : memref<?x?xf32, 1>, memref<?x?xf32>
----------------
arnab-polymage wrote:

Actually enforcing this check in verifier is causing problems with [this](https://github.com/llvm/llvm-project/blob/main/mlir/test/Dialect/SparseTensor/GPU/gpu_matvec_lib.mlir#L24) test

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