Skip to content
Snippets Groups Projects

Fixed integer square root

Merged Markus Holzer requested to merge holzer/pystencils:IntegerSquareRoot into master
2 files
+ 19
4
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -444,7 +444,7 @@ class CustomSympyPrinter(CCodePrinter):
def _print_Pow(self, expr):
"""Don't use std::pow function, for small integer exponents, write as multiplication"""
if not expr.free_symbols:
return self._typed_number(expr.evalf(), get_type_of_expression(expr))
return self._typed_number(expr.evalf(), get_type_of_expression(expr.base))
if expr.exp.is_integer and expr.exp.is_number and 0 < expr.exp < 8:
return f"({self._print(sp.Mul(*[expr.base] * expr.exp, evaluate=False))})"
Loading