diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index b7ddc393..afbe7ce9 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -423,7 +423,9 @@ def _record_leaf_ary_in_dict( self._dag_transform_cache[normalized_expr]) assert len(pt_prg.bound_arguments) == 0 - evt, out_dict = pt_prg(self.queue, **bound_arguments) + evt, out_dict = pt_prg(self.queue, + allocator=self.allocator, + **bound_arguments) evt.wait() assert len(set(out_dict) & set(key_to_frozen_subary)) == 0