Handle Tensor.__deepcopy__ via clone(), on IPU

pytorch · Nov 23, 2022 · 8e26012 · 8e26012
1 parent 7594e04
commit 8e26012
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch/_tensor.py b/torch/_tensor.py
@@ -97,7 +97,8 @@ def __deepcopy__(self, memo):
             # Update the test in test_serialization if you remove 'meta' from here
             if (
                 self.is_sparse
-                or self.device.type in ["lazy", "xla", "mps", "ort", "meta", "hpu"]
+                or self.device.type
+                in ["lazy", "xla", "mps", "ort", "meta", "hpu", "ipu"]
                 or (
                     not torch._C._has_storage(self)
                     and self.device.type == "privateuseone"