We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4d429f8 commit 69e2133Copy full SHA for 69e2133
arraycontext/impl/cupy/__init__.py
@@ -126,6 +126,8 @@ def call_loopy(
126
127
def freeze(self, array):
128
import cupy as cp
129
+ # Note that we could use a non-blocking version of cp.asnumpy here, but
130
+ # it appears to have very little impact on performance.
131
return with_array_context(rec_map_array_container(cp.asnumpy, array), actx=None)
132
133
def thaw(self, array):
0 commit comments