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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

// RUN: mlir-opt %s -test-func-set-type -split-input-file | FileCheck %s --dump-input=fail

// It's currently not possible to have an attribute with a function type due to
// parser ambiguity. So instead we reference a function declaration to take the
// type from.

// -----

// Test case: The setType call needs to erase some arg attrs.

// CHECK: func @erase_arg(f32 {test.A})
// CHECK-NOT: attributes{{.*arg[0-9]}}
func @t(f32)
func @erase_arg(%arg0: f32 {test.A}, %arg1: f32 {test.B})
attributes {test.set_type_from = @t}

// -----

// Test case: The setType call needs to erase some result attrs.

// CHECK: func @erase_result() -> (f32 {test.A})
// CHECK-NOT: attributes{{.*result[0-9]}}
func @t() -> (f32)
func @erase_result() -> (f32 {test.A}, f32 {test.B})
attributes {test.set_type_from = @t}