Skip to content

Commit 69e2133

Browse files
add a comment on non-blocking cp.asnumpy
1 parent 4d429f8 commit 69e2133

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

arraycontext/impl/cupy/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,8 @@ def call_loopy(
126126

127127
def freeze(self, array):
128128
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.
129131
return with_array_context(rec_map_array_container(cp.asnumpy, array), actx=None)
130132

131133
def thaw(self, array):

0 commit comments

Comments
 (0)