Merge branch 'main' of github.com:Tractables/pyjuice into main

Tractables · Aug 22, 2024 · ec59bd4 · ec59bd4
2 parents b4ec465 + 679708d
commit ec59bd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pyjuice/model/tensorcircuit.py b/src/pyjuice/model/tensorcircuit.py
@@ -735,7 +735,7 @@ def _init_buffer(self, name: str, shape: Tuple, set_value: Optional[float] = Non
             if not flag and tensor.size(i) != d:
                 flag = True
 
-        if not flag and check_device and tensor.device != self.device:
+        if not flag and check_device and self.device.index is not None and tensor.device != self.device:
             flag = True
 
         if flag: