From b5feb0610f9ceb18fed75cd24042cdcd0efaf47d Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sun, 7 Aug 2022 22:09:18 -0500 Subject: [PATCH] PytatoPyOpenCLArrayContext.freeze: pass allocator to loopy exec --- arraycontext/impl/pytato/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index b7ddc39..afbe7ce 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -423,7 +423,9 @@ class PytatoPyOpenCLArrayContext(_BasePytatoArrayContext): 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 -- GitLab