Disable checkpoint tests with torch compile

ghstack-source-id: 9f325224a8d5e96894449af28d61e2d7f48b124c Pull Request resolved: fairinternal/xformers#1069 __original_commit__ = fairinternal/xformers@72cefea
robandpdx-org · Mar 28, 2024 · f8b2796 · f8b2796
1 parent cf9098f
commit f8b2796
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_checkpoint.py b/tests/test_checkpoint.py
@@ -329,6 +329,7 @@ def test_optimal_checkpoint_policy(
 
 
 @pytest.mark.skipif(torch.__version__ < "2.3", reason="Only new PyTorch supported")
+@pytest.mark.skipif(True, reason="TODO[fmassa]: Broken on nightly")
 @cuda_only
 @pytest.mark.parametrize("no_grad", [False, True])
 @pytest.mark.parametrize("device", ["cuda"])