mlir/test/Conversion/StandardToLLVM/convert-argattrs.mlir

   1 // RUN: mlir-opt -convert-std-to-llvm %s | FileCheck %s
   2
   3 // CHECK-LABEL: func @check_attributes
   4 // When expanding the memref to multiple arguments, argument attributes are replicated.
   5 // CHECK-COUNT-7: {dialect.a = true, dialect.b = 4 : i64}
   6 func @check_attributes(%static: memref<10x20xf32> {dialect.a = true, dialect.b = 4 : i64 }) {
   7   return
   8 }
   9
  10 // CHECK-LABEL: func @check_multiple
  11 // Make sure arguments attributes are attached to the right argument. We match
  12 // commas in the argument list for this purpose.
  13 // CHECK: %{{.*}}: !llvm{{.*}} {first.arg = true}, %{{.*}}: !llvm{{.*}} {first.arg = true}, %{{.*}}: i{{.*}} {first.arg = true},
  14 // CHECK-SAME: %{{.*}}: !llvm{{.*}} {second.arg = 42 : i32}, %{{.*}}: !llvm{{.*}} {second.arg = 42 : i32}, %{{.*}}: i{{.*}} {second.arg = 42 : i32})
  15 func @check_multiple(%first: memref<f32> {first.arg = true}, %second: memref<f32> {second.arg = 42 : i32}) {
  16   return
  17 }