Add missing import in LBFGS cupy

ptycho · Sep 10, 2024 · 8ecccf2 · 8ecccf2
1 parent 8147dfe
commit 8ecccf2
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/ptypy/custom/LBFGS_cupy.py b/ptypy/custom/LBFGS_cupy.py
@@ -12,10 +12,7 @@
     :license: GPLv2, see LICENSE for details.
 """
 import numpy as np
-# from pycuda import gpuarray
-# import pycuda.driver as cuda
-# import pycuda.cumath
-# from pycuda.tools import DeviceMemoryPool
+import cupy as cp
 
 from ptypy.engines import register
 from ptypy.custom.LBFGS_serial import LBFGS_serial