From f35a2540c2819e68721e03bf3f4cc0a228f16707 Mon Sep 17 00:00:00 2001 From: Arpit Sen Shrivastava Date: Fri, 24 Sep 2021 15:48:32 +0530 Subject: [PATCH 1/2] Updated cupy.util to cupy.memoize On line 274, cupy.util is deprecated, updated it to cupy.memoize. This makes the code work with newer versions of cupy. --- PF-AFN_test/models/correlation/correlation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PF-AFN_test/models/correlation/correlation.py b/PF-AFN_test/models/correlation/correlation.py index c530588..e6cdb6e 100644 --- a/PF-AFN_test/models/correlation/correlation.py +++ b/PF-AFN_test/models/correlation/correlation.py @@ -271,7 +271,7 @@ def cupy_kernel(strFunction, objVariables): return strKernel # end -@cupy.util.memoize(for_each_device=True) +@cupy.memoize(for_each_device=True) def cupy_launch(strFunction, strKernel): return cupy.cuda.compile_with_cache(strKernel).get_function(strFunction) # end @@ -402,4 +402,4 @@ def __init__(self): def forward(self, tenFirst, tenSecond, intStride): return _FunctionCorrelation.apply(tenFirst, tenSecond, intStride) # end -# end \ No newline at end of file +# end From de171e63a13194d2c184154fd8f3638d9266f112 Mon Sep 17 00:00:00 2001 From: Arpit Sen Shrivastava Date: Sat, 9 Oct 2021 19:05:39 +0530 Subject: [PATCH 2/2] Update correlation.py --- PF-AFN_train/models/correlation/correlation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PF-AFN_train/models/correlation/correlation.py b/PF-AFN_train/models/correlation/correlation.py index c530588..e6cdb6e 100644 --- a/PF-AFN_train/models/correlation/correlation.py +++ b/PF-AFN_train/models/correlation/correlation.py @@ -271,7 +271,7 @@ def cupy_kernel(strFunction, objVariables): return strKernel # end -@cupy.util.memoize(for_each_device=True) +@cupy.memoize(for_each_device=True) def cupy_launch(strFunction, strKernel): return cupy.cuda.compile_with_cache(strKernel).get_function(strFunction) # end @@ -402,4 +402,4 @@ def __init__(self): def forward(self, tenFirst, tenSecond, intStride): return _FunctionCorrelation.apply(tenFirst, tenSecond, intStride) # end -# end \ No newline at end of file +# end