add todo

lhoenig · lhoenig · commit 7253344a4e04 · 2022-05-26T17:43:03.000+02:00
diff --git a/test/test_mps.py b/test/test_mps.py
@@ -1323,6 +1323,7 @@ def test_binops_dtype_precedence(self):
             full_sh = (20,)
             #print('assert1')
             for val1, val2 in itertools.product(sample_vals[dtype1], sample_vals[dtype2]):
+                # TODO Test equality of dtype of result tensor, not only values
                 self.assertEqual(
                     getattr(torch.tensor(val1, dtype=dtype1, device='mps'), binop)
                            (torch.tensor(val2, dtype=dtype2, device='mps')),
