Add Elementwise trait for ReducePrecisionOp (#2063)

Co-authored-by: Tom Natan <[email protected]>
openxla · Mar 5, 2024 · b5a199f · b5a199f
1 parent 5e24843
commit b5a199f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/stablehlo/dialect/StablehloOps.td b/stablehlo/dialect/StablehloOps.td
@@ -3126,9 +3126,9 @@ def StableHLO_UniformDequantizeOp : StableHLO_UnaryElementwiseOp<"uniform_dequan
   }];
 }
 
-def StableHLO_ReducePrecisionOp :
-    StableHLO_Op<"reduce_precision", [Pure,
-    HLO_CompatibleOperandsAndResultType /*reduce_precision_c1*/]> {
+def StableHLO_ReducePrecisionOp : StableHLO_Op<"reduce_precision",
+      [Pure, Elementwise,
+       HLO_CompatibleOperandsAndResultType /*reduce_precision_c1*/]> {
   let summary = "ReducePrecision operation";
   let description = [{
     Performs element-wise conversion of `operand` to another floating-point type