24 lines
816 B
MLIR
24 lines
816 B
MLIR
|
// RUN: mlir-opt --test-constant-fold %s | FileCheck %s
|
||
|
|
|
||
|
|
// CHECK-LABEL: func @test_const
|
||
|
|
func.func @test_const(%arg0 : index) -> tensor<4xi32> {
|
||
|
|
// CHECK: tosa.const
|
||
|
|
%0 = "tosa.const"() {value = dense<[3, 0, 1, 2]> : tensor<4xi32>} : () -> tensor<4xi32>
|
||
|
|
return %0 : tensor<4xi32>
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: func @test_const_i64
|
||
|
|
func.func @test_const_i64(%arg0 : index) -> tensor<4xi64> {
|
||
|
|
// CHECK: tosa.const
|
||
|
|
%0 = "tosa.const"() {value = dense<[3, 0, 1, 2]> : tensor<4xi64>} : () -> tensor<4xi64>
|
||
|
|
return %0 : tensor<4xi64>
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: func @try_fold_equal_with_unranked_tensor
|
||
|
|
func.func @try_fold_equal_with_unranked_tensor(%arg0: tensor<4xi32>, %arg1: tensor<i32>) {
|
||
|
|
// CHECK: tosa.equal
|
||
|
|
// CHECK-NEXT: return
|
||
|
|
%0 = tosa.equal %arg0, %arg1 : (tensor<4xi32>, tensor<i32>) -> tensor<*xi1>
|
||
|
|
return
|
||
|
|
}
|