| // RUN: mlir-opt %s -test-return-type -split-input-file -verify-diagnostics | FileCheck %s --dump-input-on-failure |
| |
| // CHECK-LABEL: testCreateFunctions |
| // This function tests invoking the create method with different inference |
| // methods. The attributes of the ops inside are used to test creation. |
| func @testCreateFunctions(%arg0 : tensor<10xf32>, %arg1 : tensor<20xi32>) { |
| // CHECK: "test.no_attributes" |
| %good = "test.no_attributes"(%arg0, %arg0) : (tensor<10xf32>, tensor<10xf32>) -> tensor<10xf32> |
| // CHECK: "test.op_with_shaped_type_infer_type_if" |
| // CHECK-SAME: (tensor<10xf32>, tensor<10xf32>) -> tensor<10x10xi17> |
| // CHECK: "test.op_with_shaped_type_infer_type_if" |
| // CHECK-SAME: (tensor<10xf32>, tensor<20xi32>) -> tensor<10x20xi17> |
| // CHECK: "test.op_with_shaped_type_infer_type_if" |
| // CHECK-SAME: (tensor<20xi32>, tensor<10xf32>) -> tensor<20x10xi17> |
| // CHECK: "test.op_with_shaped_type_infer_type_if" |
| // CHECK-SAME: (tensor<20xi32>, tensor<20xi32>) -> tensor<20x20xi17> |
| // CHECK: "test.op_with_infer_type_if" |
| // CHECK-SAME: (tensor<10xf32>, tensor<10xf32>) -> tensor<10xf32> |
| // CHECK: "test.op_with_infer_type_if" |
| // CHECK-SAME: (tensor<20xi32>, tensor<20xi32>) -> tensor<20xi32> |
| return |
| } |
| |
| // ----- |
| |
| func @testReturnTypeOpInterface(%arg0 : tensor<10xf32>) { |
| // expected-error@+1 {{incompatible with return type}} |
| %bad = "test.op_with_infer_type_if"(%arg0, %arg0) : (tensor<10xf32>, tensor<10xf32>) -> tensor<*xf32> |
| return |
| } |
| |
| // ----- |
| |
| func @testReturnTypeOpInterfaceMismatch(%arg0 : tensor<10xf32>, %arg1 : tensor<20xf32>) { |
| // expected-error@+1 {{operand type mismatch}} |
| %bad = "test.op_with_infer_type_if"(%arg0, %arg1) : (tensor<10xf32>, tensor<20xf32>) -> tensor<*xf32> |
| return |
| } |
