From 957be2f5fb234bd3bf6a15c83099ba7b1e5bcb25 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Thu, 25 Nov 2021 19:59:40 -0600
Subject: [PATCH] PytatoPyOpenCLArrayContext: always invoke MPMS materializer
---
arraycontext/impl/pytato/__init__.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py
index a7a0b1b..108079a 100644
--- a/arraycontext/impl/pytato/__init__.py
+++ b/arraycontext/impl/pytato/__init__.py
@@ -200,6 +200,10 @@ class PytatoPyOpenCLArrayContext(ArrayContext):
:arg dag: An instance of :class:`pytato.DictOfNamedArrays`
:returns: A transformed version of *dag*.
"""
+ import pytato as pt
+
+ dag = pt.transform.materialize_with_mpms(dag)
+
return dag
def tag(self, tags: Union[Sequence[Tag], Tag], array):
--
GitLab