From 04906ee2cb7e49fb18c1342025b27264be0a397e Mon Sep 17 00:00:00 2001 From: mcDandy Date: Fri, 13 Feb 2026 23:54:11 +0100 Subject: [PATCH] fix crashes --- more_math/Parser/UnifiedMathVisitor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/more_math/Parser/UnifiedMathVisitor.py b/more_math/Parser/UnifiedMathVisitor.py index bef7d9c..8961e2a 100644 --- a/more_math/Parser/UnifiedMathVisitor.py +++ b/more_math/Parser/UnifiedMathVisitor.py @@ -308,17 +308,17 @@ class UnifiedMathVisitor(MathExprVisitor): def visitMulExp(self, ctx): a = yield ctx.mulExpr() - b = yield ctx.powExpr() + b = yield ctx.shiftExpr() return self._bin_op(a, b, torch.mul, lambda a, b: a * b) def visitDivExp(self, ctx): a = yield ctx.mulExpr() - b = yield ctx.powExpr() + b = yield ctx.shiftExpr() return self._bin_op(a, b, torch.div, lambda a, b: a / b) def visitModExp(self, ctx): a = yield ctx.mulExpr() - b = yield ctx.powExpr() + b = yield ctx.shiftExpr() return self._bin_op(a, b, torch.remainder, lambda a, b: a % b) def visitPowExp(self, ctx):