Changing the order of commands so that GPU device is assigned before …

…libhandle creation.
CQCL · Apr 3, 2024 · a47d1db · a47d1db
1 parent 25e591d
commit a47d1db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytket/extensions/cutensornet/structured_state/general.py b/pytket/extensions/cutensornet/structured_state/general.py
@@ -54,7 +54,6 @@ class CuTensorNetHandle:
     """
 
     def __init__(self, device_id: Optional[int] = None):
-        self.handle = cutn.create()
         self._is_destroyed = False
 
         # Make sure CuPy uses the specified device
@@ -63,6 +62,8 @@ def __init__(self, device_id: Optional[int] = None):
         dev = cp.cuda.Device()
         self.device_id = int(dev)
 
+        self.handle = cutn.create()
+
     def __enter__(self) -> CuTensorNetHandle:
         return self