We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9ab0849 commit b5feb06Copy full SHA for b5feb06
arraycontext/impl/pytato/__init__.py
@@ -423,7 +423,9 @@ def _record_leaf_ary_in_dict(
423
self._dag_transform_cache[normalized_expr])
424
425
assert len(pt_prg.bound_arguments) == 0
426
- evt, out_dict = pt_prg(self.queue, **bound_arguments)
+ evt, out_dict = pt_prg(self.queue,
427
+ allocator=self.allocator,
428
+ **bound_arguments)
429
evt.wait()
430
assert len(set(out_dict) & set(key_to_frozen_subary)) == 0
431
0 commit comments