add cupy support

qwrapper/hamiltonian.py

@@ -1,8 +1,14 @@
+import logging
+
 import numpy
 
 from qwrapper.obs import Hamiltonian
 from qwrapper.operator import ControllablePauli
-from numpy.linalg import eigh
+
+try:
+    import cupy as np
+except ModuleNotFoundError:
+    import numpy as np
 
 
 class HeisenbergModel(Hamiltonian):
@@ -67,4 +73,4 @@ def to_matrix_hamiltonian(hamiltonian: Hamiltonian):
 
 
 def compute_ground_state(hamiltonian: Hamiltonian):
-    return min(eigh(to_matrix_hamiltonian(hamiltonian))[0])
+    return min(np.linalg.eigh(to_matrix_hamiltonian(hamiltonian))[0])

qwrapper/obs.py

@@ -2,9 +2,13 @@
 from abc import abstractmethod
 from qwrapper.circuit import QWrapper
 from qwrapper.util import QUtil
-from qulacs import QuantumState, Observable
+from qulacs import Observable
 from qwrapper.circuit import QulacsCircuit
-import numpy as np
+
+try:
+    import cupy as np
+except ModuleNotFoundError:
+    import numpy as np
 
 
 def build_operator_str(p_string):

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="qwrapper",
-    version="0.4.24",
+    version="0.4.25",
     author="kouhei nakaji",
     author_email="[email protected]",
     description="You can receive the message 'Hello!!!'",