| // RUN: mlir-opt %s --test-fold-type-converting-op --split-input-file | FileCheck %s |
| |
| // CHECK-LABEL: @test_fold_unary_op_f32_to_si32( |
| func.func @test_fold_unary_op_f32_to_si32() -> tensor<4x2xsi32> { |
| // CHECK-NEXT: %[[POSITIVE_ONE:.*]] = arith.constant dense<1> : tensor<4x2xsi32> |
| // CHECK-NEXT: return %[[POSITIVE_ONE]] : tensor<4x2xsi32> |
| %operand = arith.constant dense<5.1> : tensor<4x2xf32> |
| %sign = test.sign %operand : (tensor<4x2xf32>) -> tensor<4x2xsi32> |
| return %sign : tensor<4x2xsi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: @test_fold_binary_op_f32_to_i1( |
| func.func @test_fold_binary_op_f32_to_i1() -> tensor<8xi1> { |
| // CHECK-NEXT: %[[FALSE:.*]] = arith.constant dense<false> : tensor<8xi1> |
| // CHECK-NEXT: return %[[FALSE]] : tensor<8xi1> |
| %lhs = arith.constant dense<5.1> : tensor<8xf32> |
| %rhs = arith.constant dense<4.2> : tensor<8xf32> |
| %less_than = test.less_than %lhs, %rhs : (tensor<8xf32>, tensor<8xf32>) -> tensor<8xi1> |
| return %less_than : tensor<8xi1> |
| } |
