Add kPower case to IsNonNegative

openxla · Jun 25, 2024 · 65e79cd · 65e79cd
1 parent 61b0032
commit 65e79cd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xla/service/algebraic_simplifier.cc b/xla/service/algebraic_simplifier.cc
@@ -229,6 +229,9 @@ bool IsNonNegative(const HloInstruction* hlo,
       return IsNonNegative(hlo->operand(0), options) ||
              IsNonNegative(hlo->operand(1), options);
     }
+    case HloOpcode::kPower: {
+      return IsNonNegative(hlo->operand(0), options);
+    }
     case HloOpcode::kSelect: {
       return IsNonNegative(hlo->operand(1), options) &&
              IsNonNegative(hlo->operand(2), options);