Merge pull request #2796 from AI-Hypercomputer:chengnuojin-skip-aot

PiperOrigin-RevId: 841983264

Author: Google-ML-Automation

.github/workflows/run_tests_against_package.yml

@@ -111,4 +111,4 @@ jobs:
           fi
           # TODO: Fix the skipped tests and remove the deselect flags
           [ "${{ inputs.total_workers }}" -gt 1 ] && .venv/bin/python3 -m pip install --quiet pytest-split && SPLIT_ARGS="--splits ${{ inputs.total_workers }} --group ${{ inputs.worker_group }}" || SPLIT_ARGS=""
-          .venv/bin/python3 -m pytest ${{ inputs.pytest_addopts }} -v -m "${FINAL_PYTEST_MARKER}" --durations=0 --deselect "tests/aot_hlo_identical_test.py::AotHloIdenticalTest::test_default_hlo_match" --deselect "tests/tokenizer_test.py::TokenizerTest::test_detokenize" $SPLIT_ARGS
+          .venv/bin/python3 -m pytest ${{ inputs.pytest_addopts }} -v -m "${FINAL_PYTEST_MARKER}" --durations=0 --deselect "tests/tokenizer_test.py::TokenizerTest::test_detokenize" $SPLIT_ARGS

.github/workflows/run_tests_internal.yml

@@ -81,4 +81,4 @@ jobs:
           python3 -m pip install -e . --no-dependencies
           [ "${{ inputs.total_workers }}" -gt 1 ] && python3 -m pip install --quiet pytest-split && SPLIT_ARGS="--splits ${{ inputs.total_workers }} --group ${{ inputs.worker_group }}" || SPLIT_ARGS=""
           export LIBTPU_INIT_ARGS='--xla_tpu_scoped_vmem_limit_kib=65536'
-          python3 -m pytest ${{ inputs.pytest_addopts }} -v -m "${FINAL_PYTEST_MARKER}" --durations=0 --deselect "tests/aot_hlo_identical_test.py::AotHloIdenticalTest::test_default_hlo_match" $SPLIT_ARGS
+          python3 -m pytest ${{ inputs.pytest_addopts }} -v -m "${FINAL_PYTEST_MARKER}" --durations=0 $SPLIT_ARGS

tests/aot_hlo_identical_test.py

@@ -153,16 +153,19 @@ def assert_compile_and_real_match_hlo(self, test_name, *extra_args):
     print("AOT Compiled and train HLO files are identical for test {test_name}!")
 
   @pytest.mark.tpu_only
+  @pytest.mark.skip(reason="FileNotFoundError: /tmp/aot_test_dump. Skipped until fixing b/463839714.")
   def test_default_hlo_match(self):
     self.assert_compile_and_real_match_hlo("default_run")
 
   @pytest.mark.tpu_only
   @pytest.mark.scheduled_only
+  @pytest.mark.skip(reason="FileNotFoundError: /tmp/aot_test_dump. Skipped until fixing b/463839714.")
   def test_int8_hlo_match(self):
     self.assert_compile_and_real_match_hlo("int8", "quantization=int8")
 
   @pytest.mark.tpu_only
   @pytest.mark.scheduled_only
+  @pytest.mark.skip(reason="FileNotFoundError: /tmp/aot_test_dump. Skipped until fixing b/463839714.")
   def test_llama2_7b_hlo_match(self):
     self.assert_compile_and_real_match_hlo(
         "llama2-7b",