Merge pull request #1686 from AI-Hypercomputer:disable_llama4

maxtext authors · maxtext authors · commit 727214966694 · 2025-05-05T14:25:45.000-07:00
PiperOrigin-RevId: 755073745
diff --git a/MaxText/tests/train_compile_test.py b/MaxText/tests/train_compile_test.py
@@ -559,6 +559,7 @@ def test_moe_deepseek_without_device_limit(self):
         )
     )
 
+  @pytest.mark.skip(reason="b/415132665: Enable it once scan is supported in training for shorter compiler time")
   @pytest.mark.tpu_only
   def test_moe_llama4_17b_16e(self):
     compiled_trainstep_file = "/tmp/test_moe_llama4_17b_16e.pickle"

Original file line number	Diff line number	Diff line change
`@@ -559,6 +559,7 @@ def test_moe_deepseek_without_device_limit(self):`
`559`	`559`	`)`
`560`	`560`	`)`
`561`	`561`
	`562`	`+ @pytest.mark.skip(reason="b/415132665: Enable it once scan is supported in training for shorter compiler time")`
`562`	`563`	`@pytest.mark.tpu_only`
`563`	`564`	`def test_moe_llama4_17b_16e(self):`
`564`	`565`	`compiled_trainstep_file = "/tmp/test_moe_llama4_17b_16e.pickle"`