From 8b627d735cdf3f4b2d134b654325272c67ac4402 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Sun, 1 May 2022 12:55:12 -0500
Subject: [PATCH] to_numpy: thaw(freeze(...)) first
---
arraycontext/container/traversal.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py
index 5866ab8..a53cd5d 100644
--- a/arraycontext/container/traversal.py
+++ b/arraycontext/container/traversal.py
@@ -880,7 +880,10 @@ def to_numpy(ary: ArrayOrContainerT, actx: ArrayContext) -> Any:
f"array of type '{type(subary).__name__}' not in "
f"supported types {actx.array_types}")
- return rec_map_array_container(_to_numpy_with_check, ary)
+ return rec_map_array_container(_to_numpy_with_check,
+ # do a freeze first, if 'actx' supports
+ # container-wide freezes
+ thaw(freeze(ary, actx), actx))
# }}}
--
GitLab