Skip to content

Commit ec59bd4

Browse files
committed
Merge branch 'main' of github.com:Tractables/pyjuice into main
2 parents b4ec465 + 679708d commit ec59bd4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pyjuice/model/tensorcircuit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -735,7 +735,7 @@ def _init_buffer(self, name: str, shape: Tuple, set_value: Optional[float] = Non
735735
if not flag and tensor.size(i) != d:
736736
flag = True
737737

738-
if not flag and check_device and tensor.device != self.device:
738+
if not flag and check_device and self.device.index is not None and tensor.device != self.device:
739739
flag = True
740740

741741
if flag:

0 commit comments

Comments
 (0)