mlir/test/IR/test-func-set-type.mlir

   1 // RUN: mlir-opt %s -test-func-set-type -split-input-file | FileCheck %s
   2
   3 // It's currently not possible to have an attribute with a function type due to
   4 // parser ambiguity. So instead we reference a function declaration to take the
   5 // type from.
   6
   7 // -----
   8
   9 // Test case: The setType call needs to erase some arg attrs.
  10
  11 // CHECK: func private @erase_arg(f32 {test.A})
  12 func.func private @t(f32)
  13 func.func private @erase_arg(%arg0: f32 {test.A}, %arg1: f32 {test.B})
  14 attributes {test.set_type_from = @t}
  15
  16 // -----
  17
  18 // Test case: The setType call needs to erase some result attrs.
  19
  20 // CHECK: func private @erase_result() -> (f32 {test.A})
  21 func.func private @t() -> (f32)
  22 func.func private @erase_result() -> (f32 {test.A}, f32 {test.B})
  23 attributes {test.set_type_from = @t}