diff --git a/CLAUDE.md b/CLAUDE.md new file mode 100644 index 000000000000..dcdf409e7314 --- /dev/null +++ b/CLAUDE.md @@ -0,0 +1,15 @@ +# Testing + +Use our test class and test runner: + +``` +from torch.testing._internal.common_utils import run_tests, TestCase + +class TestFeature(TestCase): + ... + +if __name__ == "__main__": + run_tests() +``` + +To test Tensor equality, use assertEqual.