to_numpy: thaw(freeze(...)) first

inducer · May 2, 2022 · 8818204 · 8818204
1 parent 046bb52
commit 8818204
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py
@@ -845,7 +845,10 @@ def _to_numpy_with_check(subary: Any) -> 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))
 
 # }}}