From 64d639b5d87a3a81d50098c4c1cb09ca1b46c55a Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Thu, 18 Nov 2021 00:33:19 -0600
Subject: [PATCH] make dict of named arrays only once
---
arraycontext/impl/pytato/compile.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/arraycontext/impl/pytato/compile.py b/arraycontext/impl/pytato/compile.py
index 3805db2..b98a2ad 100644
--- a/arraycontext/impl/pytato/compile.py
+++ b/arraycontext/impl/pytato/compile.py
@@ -193,8 +193,7 @@ class LazilyCompilingFunctionCaller:
import loopy as lp
with ProcessLogger(logger, "transform_dag"):
- pt_dict_of_named_arrays = self.actx.transform_dag(
- pt.make_dict_of_named_arrays(dict_of_named_arrays))
+ pt_dict_of_named_arrays = self.actx.transform_dag(dict_of_named_arrays)
with ProcessLogger(logger, "generate_loopy"):
pytato_program = pt.generate_loopy(pt_dict_of_named_arrays,
@@ -280,9 +279,8 @@ class LazilyCompilingFunctionCaller:
rec_keyed_map_array_container(_as_dict_of_named_arrays,
output_template)
- from pytato import DictOfNamedArrays
compiled_func = self._dag_to_compiled_func(
- DictOfNamedArrays(dict_of_named_arrays),
+ pt.make_dict_of_named_arrays(dict_of_named_arrays),
input_id_to_name_in_program=input_id_to_name_in_program,
output_id_to_name_in_program=output_id_to_name_in_program,
output_template=output_template)
--
GitLab