| func.func @narrow_int_i32_i8() { |
| %input = util.unfoldable_constant dense<[-42, 0, 42]> : tensor<3xi32> |
| %res = "stablehlo.convert"(%input) : (tensor<3xi32>) -> tensor<3xi8> |
| check.expect_eq_const(%res, dense<[-42, 0, 42]> : tensor<3xi8>) : tensor<3xi8> |
| return |
| } |
| |
| func.func @widen_int_i8_i32() { |
| %input = util.unfoldable_constant dense<[-42, 0, 42]> : tensor<3xi8> |
| %res = "stablehlo.convert"(%input) : (tensor<3xi8>) -> tensor<3xi32> |
| check.expect_eq_const(%res, dense<[-42, 0, 42]> : tensor<3xi32>) : tensor<3xi32> |
| return |
| } |
| |
| func.func @narrow_int_i32_i16() { |
| %input = util.unfoldable_constant dense<[-42, 0, 42]> : tensor<3xi32> |
| %res = "stablehlo.convert"(%input) : (tensor<3xi32>) -> tensor<3xi16> |
| check.expect_eq_const(%res, dense<[-42, 0, 42]> : tensor<3xi16>) : tensor<3xi16> |
| return |
| } |
| |
| func.func @widen_int_i16_i32() { |
| %input = util.unfoldable_constant dense<[-42, 0, 42]> : tensor<3xi16> |
| %res = "stablehlo.convert"(%input) : (tensor<3xi16>) -> tensor<3xi32> |
| check.expect_eq_const(%res, dense<[-42, 0, 42]> : tensor<3xi32>) : tensor<3xi32> |
| return |
| } |
| |
| func.func @narrow_int_i64_i32() { |
| %input = util.unfoldable_constant dense<[-42, 0, 42]> : tensor<3xi64> |
| %res = "stablehlo.convert"(%input) : (tensor<3xi64>) -> tensor<3xi32> |
| check.expect_eq_const(%res, dense<[-42, 0, 42]> : tensor<3xi32>) : tensor<3xi32> |
| return |
| } |
| |
| func.func @widen_int_i32_i64() { |
| %input = util.unfoldable_constant dense<[-42, 0, 42]> : tensor<3xi32> |
| %res = "stablehlo.convert"(%input) : (tensor<3xi32>) -> tensor<3xi64> |
| check.expect_eq_const(%res, dense<[-42, 0, 42]> : tensor<3xi64>) : tensor<3xi64> |
| return |
| } |
| |
| func.func @int_to_float() { |
| %input = util.unfoldable_constant dense<[-42, 0, 42]> : tensor<3xi32> |
| %res = "stablehlo.convert"(%input) : (tensor<3xi32>) -> tensor<3xf32> |
| check.expect_almost_eq_const(%res, dense<[-42.0, 0.0, 42.0]> : tensor<3xf32>) : tensor<3xf32> |
| return |
| } |
| |
| // TODO(#6160): XLA does not specify the rounding behavior, meaning that we |
| // can't test something like -10.5 as that could be -11 (roundf) or -10 (rint |
| // with round-to-even mode). |
| // |
| // For casting rules, see |
| // https://www.tensorflow.org/xla/operation_semantics#convertelementtype |
| // func.func @float_to_int() { |
| // %input = util.unfoldable_constant dense<[-10.5, -4.4, 4.4, 10.5]> : tensor<4xf32> |
| // %res = "stablehlo.convert"(%input) : (tensor<4xf32>) -> tensor<4xi32> |
| // check.expect_eq_const(%res, dense<[-10, -4, 4, 10]> : tensor<4xi32>) : tensor<4xi32> |
| // return |
| // } |
