mlir/test/IR/result.mlir

   1 // RUN: mlir-opt %s -split-input-file -verify-diagnostics | FileCheck %s
   2
   3 //===----------------------------------------------------------------------===//
   4 // Test mixed normal and variadic results
   5 //===----------------------------------------------------------------------===//
   6
   7 func.func @correct_variadic_result() -> tensor<f32> {
   8   // CHECK: mixed_normal_variadic_result
   9   %0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, tensor<f32>, tensor<f32>, tensor<f32>, tensor<f32>)
  10   return %0#4 : tensor<f32>
  11 }
  12
  13 // -----
  14
  15 func.func @error_in_first_variadic_result() -> tensor<f32> {
  16   // expected-error @+1 {{result #1 must be variadic of tensor of any type}}
  17   %0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, f32, tensor<f32>, tensor<f32>, tensor<f32>)
  18   return %0#4 : tensor<f32>
  19 }
  20
  21 // -----
  22
  23 func.func @error_in_normal_result() -> tensor<f32> {
  24   // expected-error @+1 {{result #2 must be tensor of any type}}
  25   %0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, tensor<f32>, f32, tensor<f32>, tensor<f32>)
  26   return %0#4 : tensor<f32>
  27 }
  28
  29 // -----
  30
  31 func.func @error_in_second_variadic_result() -> tensor<f32> {
  32   // expected-error @+1 {{result #3 must be variadic of tensor of any type}}
  33   %0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, tensor<f32>, tensor<f32>, f32, tensor<f32>)
  34   return %0#4 : tensor<f32>
  35 }
  36