24 lines · plain
1// RUN: mlir-opt %s -test-func-set-type -split-input-file | FileCheck %s2 3// It's currently not possible to have an attribute with a function type due to4// parser ambiguity. So instead we reference a function declaration to take the5// 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})12func.func private @t(f32)13func.func private @erase_arg(%arg0: f32 {test.A}, %arg1: f32 {test.B})14attributes {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})21func.func private @t() -> (f32)22func.func private @erase_result() -> (f32 {test.A}, f32 {test.B})23attributes {test.set_type_from = @t}24