Skip to content

Commit 60aa77e

Browse files
committed
Fixing minor issue.
1 parent c480103 commit 60aa77e

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

hist/exact/basic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ def __count_votes(im, out, size, invert):
349349

350350
if HAS_CUPY:
351351
import cupy # pylint: disable=import-error
352-
@cupy.util.memoize(for_each_device=True)
352+
@cupy.memoize(for_each_device=True)
353353
def __get_count_votes_cupy_kernel(invert):
354354
sign = '>' if invert else '<'
355355
return cupy.ReductionKernel('X x', 'Y y',

hist/exact/va.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def __calculate_d_Phi(u, t, G_info, alpha_2, d_phi, denom): # pylint: disable=to
180180

181181
if HAS_CUPY:
182182
import cupy # pylint: disable=import-error
183-
@cupy.util.memoize(for_each_device=True)
183+
@cupy.memoize(for_each_device=True)
184184
def __get_cupy_kernel(ndim, niters=5):
185185
"""
186186
Create the cupy ElementwiseKernel for the VA algorithm with the given number of dimensions

0 commit comments

Comments
 (0)