diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py index 9e0992d26e3aad8cfe77a809ddb2f9d6f2b1fc86..c0218292dc8c43636ce8e7d6e0868a7b8e5024ef 100644 --- a/arraycontext/impl/pytato.py +++ b/arraycontext/impl/pytato.py @@ -96,6 +96,11 @@ class _PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace): import pytato as pt return pt.amax(a) + def stack(self, arrays, axis=0): + import pytato as pt + from meshmode.dof_array import obj_or_dof_array_vectorize_n_args + return obj_or_dof_array_vectorize_n_args(pt.stack, arrays, axis) + class PytatoCompiledOperator: def __init__(self, actx, pytato_program, input_spec, output_spec):