enable aoti for preprocess ci

[ghstack-poisoned]

Author: lucylq

examples/models/llama3_2_vision/preprocess/test_preprocess.py

@@ -76,10 +76,10 @@ def initialize_models(resize_to_max_canvas: bool) -> Dict[str, Any]:
             strict=False,
         )
 
-        # aoti_path = torch._inductor.aot_compile(
-        #     exported_model.module(),
-        #     model.get_example_inputs(),
-        # )
+        aoti_path = torch._inductor.aot_compile(
+            exported_model.module(),
+            model.get_example_inputs(),
+        )
 
         edge_program = to_edge(
             exported_model, compile_config=EdgeCompileConfig(_check_ir_validity=False)
@@ -91,7 +91,7 @@ def initialize_models(resize_to_max_canvas: bool) -> Dict[str, Any]:
             "reference_model": reference_model,
             "model": model,
             "exported_model": exported_model,
-            # "aoti_path": aoti_path,
+            "aoti_path": aoti_path,
             "executorch_model": executorch_model,
         }
 
@@ -237,11 +237,11 @@ def run_preprocess(
         self.assertEqual(reference_ar, et_ar.tolist())
 
         # Run aoti model and check it matches reference model.
-        # aoti_path = models["aoti_path"]
-        # aoti_model = torch._export.aot_load(aoti_path, "cpu")
-        # aoti_image, aoti_ar = aoti_model(image_tensor, inscribed_size, best_resolution)
-        # self.assertTrue(torch.allclose(reference_image, aoti_image))
-        # self.assertEqual(reference_ar, aoti_ar.tolist())
+        aoti_path = models["aoti_path"]
+        aoti_model = torch._export.aot_load(aoti_path, "cpu")
+        aoti_image, aoti_ar = aoti_model(image_tensor, inscribed_size, best_resolution)
+        self.assertTrue(torch.allclose(reference_image, aoti_image))
+        self.assertEqual(reference_ar, aoti_ar.tolist())
 
     # This test setup mirrors the one in torchtune:
     # https://github.com/pytorch/torchtune/blob/main/tests/torchtune/models/clip/test_clip_image_transform.py