This commit is contained in:
Disty0
2025-06-25 23:37:41 +03:00
parent e43d1d2ba7
commit c9f49720c5
3 changed files with 4 additions and 4 deletions
+1 -1
View File
@@ -665,6 +665,6 @@ def normalize_device(dev):
def same_device(d1, d2):
if d1.type != d2.type:
if torch.device(d1).type != torch.device(d2).type:
return False
return normalize_device(d1) == normalize_device(d2)