diff --git a/loopy/target/python.py b/loopy/target/python.py index 1940dab1622afe7a879235bedf2265bb5afab33e..5760b56d56140d4387a4cdd0c2f93abfae6d8aa5 100644 --- a/loopy/target/python.py +++ b/loopy/target/python.py @@ -25,6 +25,8 @@ THE SOFTWARE. from typing import Optional, Sequence, Tuple +import numpy as np + from pymbolic.mapper import Mapper from pymbolic.mapper.stringifier import StringifyMapper from genpy import Generable, Suite, Collection @@ -58,7 +60,10 @@ class ExpressionToPythonMapper(StringifyMapper): __call__ = rec def map_constant(self, expr, enclosing_prec): - return repr(expr) + if isinstance(expr, np.generic): + return repr(expr).replace("np.", "_lpy_np.") + else: + return repr(expr) def map_variable(self, expr, enclosing_prec): if expr.name in self.codegen_state.var_subst_map: