Use self.assertEqual instead of torch.testing.assert_close in tests

Replace manual tolerance specification with self.assertEqual which
automatically handles appropriate tolerances for tensor comparisons.
This makes the tests more concise and follows PyTorch testing conventions.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: soumith

test/test_extension.py

@@ -147,7 +147,7 @@ def forward(self, x):
                     compiled_model = torch.compile(model, mode="reduce-overhead", fullgraph=True)
                     actual = compiled_model(x)
 
-                torch.testing.assert_close(actual, expected, rtol=1e-5, atol=1e-5)
+                self.assertEqual(actual, expected)
 
     @unittest.skipIf(not torch.cuda.is_available(), "requires cuda")
     def test_compile_custom_only(self):
@@ -166,7 +166,7 @@ def model(x):
                     compiled_model = torch.compile(model, mode="reduce-overhead", fullgraph=True)
                     actual = compiled_model(x)
 
-                torch.testing.assert_close(actual, expected, rtol=1e-5, atol=1e-5)
+                self.assertEqual(actual, expected)
 
 
 if __name__ == "__main__":