[MPS] Fix non-contig to contig tensor copy (#86056)
This handles a rare case when MPS tensor is constructed from non-contiguous CPU tensor.
Fixes https://github.com/pytorch/pytorch/issues/85967
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86056
Approved by: https://github.com/janeyx99
diff --git a/test/test_mps.py b/test/test_mps.py
index 27d548e..3c5c919 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -1642,6 +1642,13 @@
mps_result = rotate_subset(mps_data)
self.assertEqual(cpu_result, mps_result.to("cpu"))
+ # See https://github.com/pytorch/pytorch/issues/85967
+ def test_from_numpy_non_contiguous(self):
+ a = np.arange(9).reshape(3, 3)[:, :2]
+ t_cpu = torch.tensor(a, device="cpu")
+ t_mps = torch.tensor(a, device="mps")
+ self.assertEqual(t_cpu, t_mps.to("cpu"))
+
class TestLogical(TestCase):
def _wrap_tensor(self, x, device="cpu", dtype=None, requires_grad=False):
