commit 09f2373ec0b491b38d893215ff5289c839e62d35
author Sergii Dymchenko <sdym@fb.com> Wed Nov 30 17:00:36 2022 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Wed Nov 30 17:00:36 2022 +0000
tree 8688cf8e658af7d7583ac4376f3df098cb7d893c
parent f1415b8cb6c7b5827940dcd9f230e0e66e07ab39

Fix TODOs related to #38095 in test_mps.py (#89815)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/89815
Approved by: https://github.com/weiwangmeta, https://github.com/kulinseth

test/test_mps.py

diff --git a/test/test_mps.py b/test/test_mps.py
index 044b3a7..77c2a34 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -1965,8 +1965,7 @@
 
         output_cpu = F.nll_loss(input, target, reduction=reduction)
         output_mps = F.nll_loss(input_mps, target_mps, reduction=reduction)
-        # TODO(#38095): Replace assertEqualIgnoreType. See issue #38095
-        self.assertEqualIgnoreType(output_cpu, output_mps.to('cpu'))
+        self.assertEqual(output_cpu, output_mps.to('cpu'))
 
         output_cpu.sum().backward()
         output_mps.sum().backward()
@@ -1985,8 +1984,7 @@
 
         output_cpu = F.nll_loss(input, target, reduction=reduction)
         output_mps = F.nll_loss(input_mps, target_mps, reduction=reduction)
-        # TODO(#38095): Replace assertEqualIgnoreType. See issue #38095
-        self.assertEqualIgnoreType(output_cpu, output_mps.to('cpu'))
+        self.assertEqual(output_cpu, output_mps.to('cpu'))
 
         output_cpu.sum().backward()
         output_mps.sum().backward()