Update verifier

Summary: Fixes #7998

Differential Revision: D68839524

Author: angelayi

exir/program/test/test_program.py

@@ -313,6 +313,45 @@ def forward(self, x, y):
         )
         edge_manager.to_executorch()
 
+    def test_data_dependent(self):
+        with torch.library._scoped_library("mylib", "FRAGMENT") as lib:
+            torch.library.define(
+                "mylib::foo1",
+                "(Tensor a, Tensor b) -> Tensor",
+                tags=torch.Tag.pt2_compliant_tag,
+                lib=lib,
+            )
+
+            @torch.library.impl("mylib::foo1", "cpu", lib=lib)
+            def foo_impl(a, b):
+                return a + b
+
+            @torch.library.register_fake("mylib::foo1", lib=lib)
+            def mylib_foo_default_fake(*args, **kwargs):
+                ctx = torch.library.get_ctx()
+                fake_shape = ctx.new_dynamic_size()
+                return torch.empty(fake_shape, dtype=torch.float32, device="cpu")
+
+            class M(torch.nn.Module):
+                def forward(self, a, b, c):
+                    res = torch.ops.mylib.foo1(a, b)
+
+                    c_item = c.item()
+                    torch._check_is_size(c_item)
+                    torch._check(c_item < res.shape[0])
+                    return res[:c_item]
+
+            inp = (torch.randn(10), torch.randn(10), torch.tensor(3))
+
+            ep = export(M(), inp)
+            edge = to_edge(ep)
+            self.assertTrue(torch.allclose(
+                edge.exported_program().module()(
+                    *inp
+                ),
+                M()(*inp),
+            ))
+
     def test_edge_manager_transform(self):
         edge_manager: EdgeProgramManager = to_edge(
             get_exported_programs(), get_config_methods()

exir/verification/verifier.py

@@ -20,6 +20,8 @@
     EdgeOpArgValidator,
     RunHigherOrderOperatorError,
 )
+from executorch.exir.passes.executorch_prim_ops_registry import _EXECUTORCH_SYM_OPS
+from executorch.exir.passes.replace_aten_with_edge_pass import DISALLOW_LIST
 
 from torch._dispatch.python import enable_python_dispatcher
 from torch._export.utils import _detect_fake_mode_from_gm
@@ -107,8 +109,7 @@ def _get_exception_list(self) -> List[torch._ops.OpOverload]:
                 torch.ops.aten.max.default,  # TODO(T188268054)
                 torch.ops.aten.min.default,  # TODO(T188268054)
                 torch.ops.aten.full_like.default,  # TODO(T183507359)
-            ]
-            exception_list += self._exception_list
+            ] + list(_EXECUTORCH_SYM_OPS) + DISALLOW_LIST+ self._exception_list
 
             return exception_list
 
@@ -251,11 +252,12 @@ def check_valid_edge_op(self, op):
                 op
                 in [
                     operator.getitem,
-                    torch.ops.aten.sym_size.int,
                     torch.ops.aten.scalar_tensor.default,
                     torch.ops.aten._assert_async.msg,
                     torch.ops.aten._assert_scalar.default,
                 ]
+                + DISALLOW_LIST
+                + list(_EXECUTORCH_SYM_OPS)
                 + self._exception_list
             ):
                 return