forked from OSchip/llvm-project
102 lines
5.4 KiB
MLIR
102 lines
5.4 KiB
MLIR
// RUN: mlir-opt -allow-unregistered-dialect -split-input-file -verify-diagnostics %s | FileCheck %s
|
|
|
|
// CHECK-LABEL: @source
|
|
func @source(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>, %arg2: tensor<4xf32>) -> (tensor<4xf32>) {
|
|
%0 = "quant.region"(%arg0, %arg1, %arg2) ({
|
|
^bb0(%10: tensor<4xf32>, %11: tensor<4xf32>, %12: tensor<4xf32>):
|
|
%13 = "foo"(%10, %11) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
%14 = "bar"(%13, %12) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
"quant.return"(%14) : (tensor<4xf32>) -> ()
|
|
}) {input_specs = [f32, f32, f32], output_specs = [f32], logical_kernel = "xyz"}
|
|
: (tensor<4xf32>, tensor<4xf32>, tensor<4xf32>) -> (tensor<4xf32>)
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|
|
// CHECK-LABEL: @annotated
|
|
func @annotated(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>, %arg2: tensor<4xf32>) -> (tensor<4xf32>) {
|
|
%0 = "quant.region"(%arg0, %arg1, %arg2) ({
|
|
^bb0(%10: tensor<4xf32>, %11: tensor<4xf32>, %12: tensor<4xf32>):
|
|
%13 = "foo"(%10, %11) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
%14 = "bar"(%13, %12) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
"quant.return"(%14) : (tensor<4xf32>) -> ()
|
|
}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, f32],
|
|
output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
|
|
: (tensor<4xf32>, tensor<4xf32>, tensor<4xf32>) -> (tensor<4xf32>)
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|
|
// CHECK-LABEL: @quantized
|
|
func @quantized(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>, %arg2: tensor<4xf32>) -> (tensor<4xf32>) {
|
|
%0 = "quant.region"(%arg0, %arg1, %arg2) ({
|
|
^bb0(%10: tensor<4xf32>, %11: tensor<4xf32>, %12: tensor<4xf32>):
|
|
%13 = "foo"(%10, %11) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
%14 = "bar"(%13, %12) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
"quant.return"(%14) : (tensor<4xf32>) -> ()
|
|
}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, !quant.uniform<i32:f32, 2.0>],
|
|
output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
|
|
: (tensor<4xf32>, tensor<4xf32>, tensor<4xf32>) -> (tensor<4xf32>)
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @unmatched_quantize(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>, %arg2: tensor<4xf32>) -> (tensor<4xf32>) {
|
|
// @expected-error @+1 {{'quant.region' op has incompatible specification !quant.uniform<i32:f16, 3.000000e+00> and input type 'tensor<4xf32>'}}
|
|
%0 = "quant.region"(%arg0, %arg1, %arg2) ({
|
|
^bb0(%10: tensor<4xf32>, %11: tensor<4xf32>, %12: tensor<4xf32>):
|
|
%13 = "foo"(%10, %11) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
%14 = "bar"(%13, %12) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
"quant.return"(%14) : (tensor<4xf32>) -> ()
|
|
}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, !quant.uniform<i32:f16, 3.0>],
|
|
output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
|
|
: (tensor<4xf32>, tensor<4xf32>, tensor<4xf32>) -> (tensor<4xf32>)
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @unmatched_primitive(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>, %arg2: tensor<4xf32>) -> (tensor<4xf32>) {
|
|
// @expected-error @+1 {{'quant.region' op has incompatible specification i32 and input type 'tensor<4xf32>'}}
|
|
%0 = "quant.region"(%arg0, %arg1, %arg2) ({
|
|
^bb0(%10: tensor<4xf32>, %11: tensor<4xf32>, %12: tensor<4xf32>):
|
|
%13 = "foo"(%10, %11) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
%14 = "bar"(%13, %12) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
"quant.return"(%14) : (tensor<4xf32>) -> ()
|
|
}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, i32],
|
|
output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
|
|
: (tensor<4xf32>, tensor<4xf32>, tensor<4xf32>) -> (tensor<4xf32>)
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @unmatched_number(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>, %arg2: tensor<4xf32>) -> (tensor<4xf32>) {
|
|
// @expected-error @+1 {{'quant.region' op has unmatched operands/results number and spec attributes number}}
|
|
%0 = "quant.region"(%arg0, %arg1, %arg2) ({
|
|
^bb0(%10: tensor<4xf32>, %11: tensor<4xf32>, %12: tensor<4xf32>):
|
|
%13 = "foo"(%10, %11) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
%14 = "bar"(%13, %12) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
"quant.return"(%14) : (tensor<4xf32>) -> ()
|
|
}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>],
|
|
output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
|
|
: (tensor<4xf32>, tensor<4xf32>, tensor<4xf32>) -> (tensor<4xf32>)
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @isolated(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>, %arg2: tensor<4xf32>) -> (tensor<4xf32>) {
|
|
// @expected-note @+1 {{required by region isolation constraints}}
|
|
%0 = "quant.region"(%arg0, %arg1) ({
|
|
^bb0(%10: tensor<4xf32>, %11: tensor<4xf32>):
|
|
%13 = "foo"(%10, %11) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
// @expected-error @+1 {{'bar' op using value defined outside the region}}
|
|
%14 = "bar"(%13, %arg2) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
|
|
"quant.return"(%14) : (tensor<4xf32>) -> ()
|
|
}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>],
|
|
output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
|
|
: (tensor<4xf32>, tensor<4xf32>) -> (tensor<4xf32>)
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|