Fix typo in NNAPI tests (#63797)
authorAkshit Khurana <axit@fb.com>
Mon, 23 Aug 2021 23:33:07 +0000 (16:33 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 23 Aug 2021 23:34:24 +0000 (16:34 -0700)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63797

nnapi memory format test has a typo

Test Plan:
pytest test/test_nnapi.py::TestNNAPI

Imported from OSS

Reviewed By: Amyh11325

Differential Revision: D30495473

fbshipit-source-id: 8edad7c01a080847a64a2797e077ec4d6077552a

test/test_nnapi.py

index d70bebf..f8db7e1 100644 (file)
@@ -78,7 +78,7 @@ class TestNNAPI(TestCase):
                     # to get a nice message.
                     self.assertEqual(eager_output, nnapi_output, atol=0, rtol=0)
             if expected_memory_format:
-                self.assertTrue(nnapi_out.is_contiguous(memory_format=expected_memory_format))
+                self.assertTrue(nnapi_output.is_contiguous(memory_format=expected_memory_format))
 
     def float_and_quant_and_nhwc(self, inp_float, scale, zero_point):
         torch.manual_seed(29)