Update

Author: hsuan-lun-chiang

tests/integration/aot_identical_test.py

@@ -80,6 +80,7 @@ def check_large_files_equal(self, file_path1, file_path2):
     return h1.hexdigest() == h2.hexdigest()
 
 
+@pytest.mark.integration_test
 class AotHloIdenticalTest(AotBaseTest):
   """Tests for Ahead of Time Compilation HLO Graph Verification."""
 
@@ -158,6 +159,7 @@ def test_default_hlo_match(self):
     self.assert_compile_and_real_match_hlo("default_run")
 
 
+@pytest.mark.integration_test
 class AotJaxprIdenticalTest(AotBaseTest):
   """Tests for Ahead of Time Compilation Jaxpr Verification."""