Merge branch 'main' into crpa/lookaside_autograd-function-apply

Lightning-AI · Oct 3, 2024 · b4647ed · b4647ed
2 parents a0caaea + 67d8df6
commit b4647ed
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/thunder/tests/test_grad.py b/thunder/tests/test_grad.py
@@ -1460,6 +1460,9 @@ def test_populate_grads_mlp(executor, device, dtype):
 
 @instantiate(dtypes=(thunder.float32,))
 def test_populate_grads_csa(executor, device, dtype):
+    if version_between(torch.__version__, min_ver="2.6.0a0", max_ver="2.6.0"):
+        pytest.skip("https://github.com/Lightning-AI/lightning-thunder/issues/1254")
+
     from thunder.benchmarks import NanoGPTCSABenchmark, NanoGPTConfig
 
     # NOTE Currently setting dropout to zero for reproducibility, other settings taken from gpt2 config
@@ -1487,6 +1490,9 @@ def test_populate_grads_csa(executor, device, dtype):
 
 @instantiate(dtypes=(thunder.float32,))
 def test_populate_grads_block(executor, device, dtype):
+    if version_between(torch.__version__, min_ver="2.6.0a0", max_ver="2.6.0"):
+        pytest.skip("https://github.com/Lightning-AI/lightning-thunder/issues/1254")
+
     from thunder.benchmarks import NanoGPTBlockBenchmark, NanoGPTConfig
 
     # NOTE Currently setting dropout to zero for reproducibility, other settings taken from gpt2 config