| func.func @select() { | |
| %input = util.unfoldable_constant dense<[1, 0, 1, 0]> : tensor<4xi1> | |
| %zeros = util.unfoldable_constant dense<0> : tensor<4xi1> | |
| %cond = "stablehlo.compare"(%input, %zeros) {comparison_direction = #stablehlo<comparison_direction GT>} : (tensor<4xi1>, tensor<4xi1>) -> tensor<4xi1> | |
| %lhs = util.unfoldable_constant dense<[1, 2, 3, 4]> : tensor<4xi32> | |
| %rhs = util.unfoldable_constant dense<[5, 6, 7, 8]> : tensor<4xi32> | |
| %result = "stablehlo.select"(%cond, %lhs, %rhs) : (tensor<4xi1>, tensor<4xi32>, tensor<4xi32>) -> tensor<4xi32> | |
| check.expect_eq_const(%result, dense<[1,6, 3, 8]> : tensor<4xi32>) : tensor<4xi32> | |
| return | |
| } |