forked from OSchip/llvm-project
37 lines
1.3 KiB
MLIR
37 lines
1.3 KiB
MLIR
// RUN: mlir-opt %s -split-input-file -verify-diagnostics | FileCheck %s
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test mixed normal and variadic results
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @correct_variadic_result() -> tensor<f32> {
|
|
// CHECK: mixed_normal_variadic_result
|
|
%0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, tensor<f32>, tensor<f32>, tensor<f32>, tensor<f32>)
|
|
return %0#4 : tensor<f32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @error_in_first_variadic_result() -> tensor<f32> {
|
|
// expected-error @+1 {{result #1 must be tensor of any type}}
|
|
%0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, f32, tensor<f32>, tensor<f32>, tensor<f32>)
|
|
return %0#4 : tensor<f32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @error_in_normal_result() -> tensor<f32> {
|
|
// expected-error @+1 {{result #2 must be tensor of any type}}
|
|
%0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, tensor<f32>, f32, tensor<f32>, tensor<f32>)
|
|
return %0#4 : tensor<f32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @error_in_second_variadic_result() -> tensor<f32> {
|
|
// expected-error @+1 {{result #3 must be tensor of any type}}
|
|
%0:5 = "test.mixed_normal_variadic_result"() : () -> (tensor<f32>, tensor<f32>, tensor<f32>, f32, tensor<f32>)
|
|
return %0#4 : tensor<f32>
|
|
}
|
|
|