From c50f6849a140016ec8dcf51ee3c0ddc95c4dda68 Mon Sep 17 00:00:00 2001
From: Akshay Gopinath Bhardwaj <akshaygb@meta.com>
Date: Wed, 18 Jun 2025 11:28:07 -0700
Subject: [PATCH] Add legacy mode test (#11492)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/11492

Add the missing legacy mode tests for test_div.py

Reviewed By: mcr229

Differential Revision: D76167854
---
 backends/xnnpack/test/ops/test_div.py | 50 +++++++++++++++------------
 1 file changed, 28 insertions(+), 22 deletions(-)

diff --git a/backends/xnnpack/test/ops/test_div.py b/backends/xnnpack/test/ops/test_div.py
index b53c59df8e1..3d9835ec56e 100644
--- a/backends/xnnpack/test/ops/test_div.py
+++ b/backends/xnnpack/test/ops/test_div.py
@@ -31,17 +31,20 @@ def forward(self, x):
             return z
 
     def _test_div(self, inputs):
-        (
-            Tester(self.Div(), inputs)
-            .export()
-            .check_count({"torch.ops.aten.div.Tensor": 1})
-            .to_edge_transform_and_lower()
-            .check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
-            .check_not(["executorch_exir_dialects_edge__ops_aten_div_Tensor"])
-            .to_executorch()
-            .serialize()
-            .run_method_and_compare_outputs()
-        )
+        for legacy_mode in (True, False):
+            tester = Tester(self.Div(), inputs)
+            tester.export()
+            tester.check_count({"torch.ops.aten.div.Tensor": 1})
+            if legacy_mode:
+                tester.to_edge()
+                tester.partition()
+            else:
+                tester.to_edge_transform_and_lower()
+            tester.check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
+            tester.check_not(["executorch_exir_dialects_edge__ops_aten_div_Tensor"])
+            tester.to_executorch()
+            tester.serialize()
+            tester.run_method_and_compare_outputs()
 
     def test_fp16_div(self):
         # Adding 4 to move distribution away from 0, 4 Std Dev should be far enough
@@ -59,14 +62,17 @@ def test_fp32_div(self):
     def test_fp32_div_single_input(self):
         # Adding 4 to move distribution away from 0, 4 Std Dev should be far enough
         inputs = (torch.randn(1) + 4,)
-        (
-            Tester(self.DivSingleInput(), inputs)
-            .export()
-            .check_count({"torch.ops.aten.div.Tensor": 1})
-            .to_edge_transform_and_lower()
-            .check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
-            .check_not(["executorch_exir_dialects_edge__ops_aten_div_Tensor"])
-            .to_executorch()
-            .serialize()
-            .run_method_and_compare_outputs()
-        )
+        for legacy_mode in (True, False):
+            tester = Tester(self.DivSingleInput(), inputs)
+            tester.export()
+            tester.check_count({"torch.ops.aten.div.Tensor": 1})
+            if legacy_mode:
+                tester.to_edge()
+                tester.partition()
+            else:
+                tester.to_edge_transform_and_lower()
+            tester.check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
+            tester.check_not(["executorch_exir_dialects_edge__ops_aten_div_Tensor"])
+            tester.to_executorch()
+            tester.serialize()
+            tester.run_method_and_compare_outputs()