From 37e142e7c5d3c608ea580b0861c1abaff3c2873e Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 10 Jun 2022 13:43:56 -0500
Subject: [PATCH] Array identifiers in pytato freezes: use "_ary", not "_actx"
---
arraycontext/impl/pytato/__init__.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py
index b5e7f63..f626345 100644
--- a/arraycontext/impl/pytato/__init__.py
+++ b/arraycontext/impl/pytato/__init__.py
@@ -265,7 +265,7 @@ class PytatoPyOpenCLArrayContext(_BasePytatoArrayContext):
def _record_leaf_ary_in_dict(key: Tuple[Any, ...],
ary: ArrayT):
- key_str = "_actx" + _ary_container_key_stringifier(key)
+ key_str = "_ary" + _ary_container_key_stringifier(key)
array_as_dict[key_str] = ary
return ary
@@ -340,7 +340,7 @@ class PytatoPyOpenCLArrayContext(_BasePytatoArrayContext):
}
def _to_frozen(key: Tuple[Any, ...], ary: ArrayT):
- key_str = "_actx" + _ary_container_key_stringifier(key)
+ key_str = "_ary" + _ary_container_key_stringifier(key)
return key_to_frozen_subary[key_str]
return with_array_context(rec_keyed_map_array_container(_to_frozen,
@@ -488,7 +488,7 @@ class PytatoJAXArrayContext(_BasePytatoArrayContext):
def _record_leaf_ary_in_dict(key: Tuple[Any, ...],
ary: Union[DeviceArray, pt.Array]):
- key_str = "_actx" + _ary_container_key_stringifier(key)
+ key_str = "_ary" + _ary_container_key_stringifier(key)
array_as_dict[key_str] = ary
return ary
@@ -524,7 +524,7 @@ class PytatoJAXArrayContext(_BasePytatoArrayContext):
}
def _to_frozen(key: Tuple[Any, ...], ary: ArrayT):
- key_str = "_actx" + _ary_container_key_stringifier(key)
+ key_str = "_ary" + _ary_container_key_stringifier(key)
return key_to_frozen_subary[key_str]
return with_array_context(rec_keyed_map_array_container(_to_frozen,
--
GitLab