[Mlir-commits] [mlir] [mlir][GPUToNVVM] Fix memref function args/results (PR #96392)
Matthias Springer
llvmlistbot at llvm.org
Sat Jun 22 08:46:11 PDT 2024
================
@@ -679,6 +680,28 @@ gpu.module @gpumodule {
}
+gpu.module @test_module_34 {
+ // CHECK-LABEL: llvm.func @memref_signature(
+ // CHECK-SAME: %{{.*}}: !llvm.ptr, %{{.*}}: !llvm.ptr, %{{.*}}: i64, %{{.*}}: i64, %{{.*}}: i64, %{{.*}}: f32) -> !llvm.struct<(struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>, f32)>
+ // CHECK: llvm.mlir.undef
+ // CHECK: llvm.insertvalue
+ // CHECK: llvm.insertvalue
+ // CHECK: llvm.insertvalue
+ // CHECK: llvm.insertvalue
+ // CHECK: llvm.insertvalue
+ // CHECK: llvm.mlir.undef
+ // CHECK: llvm.insertvalue
+ // CHECK: llvm.insertvalue
+ // CHECK: llvm.return
+
+ // CHECK-BARE-LABEL: llvm.func @memref_signature(
+ // CHECK-BARE-SAME: %{{.*}}: !llvm.ptr, %{{.*}}: f32) -> !llvm.struct<(ptr, f32)>
+ gpu.func @memref_signature(%m: memref<2xf32>, %f: f32) -> (memref<2xf32>, f32) {
+ gpu.return %m, %f : memref<2xf32>, f32
----------------
matthias-springer wrote:
That makes sense. We already verify that for `gpu.func` but not for `llvm.func`. See #96399.
https://github.com/llvm/llvm-project/pull/96392
More information about the Mlir-commits
mailing list