Skip to content

Commit b5feb06

Browse files
committed
PytatoPyOpenCLArrayContext.freeze: pass allocator to loopy exec
1 parent 9ab0849 commit b5feb06

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

arraycontext/impl/pytato/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,9 @@ def _record_leaf_ary_in_dict(
423423
self._dag_transform_cache[normalized_expr])
424424

425425
assert len(pt_prg.bound_arguments) == 0
426-
evt, out_dict = pt_prg(self.queue, **bound_arguments)
426+
evt, out_dict = pt_prg(self.queue,
427+
allocator=self.allocator,
428+
**bound_arguments)
427429
evt.wait()
428430
assert len(set(out_dict) & set(key_to_frozen_subary)) == 0
429431

0 commit comments

Comments
 (0)