[dynamo][export] Add some missing trace rules

ghstack-source-id: 229b3c2854b9401b07ff4a563b4da7687739daf2
Pull Request resolved: https://github.com/pytorch/pytorch/pull/164080

test/export/test_export_with_inline_and_install.py

@@ -73,22 +73,6 @@ del test
 unittest.expectedFailure(
     InlineAndInstallStrictExportTestExport.test_buffer_util_inline_and_install_strict  # noqa: F821
 )
-# this is because we marked unlift hooks to be dynamo skip traced
-unittest.expectedFailure(
-    InlineAndInstallStrictExportTestExport.test_custom_tag_metadata_re_export_inline_and_install_strict  # noqa: F821
-)
-unittest.expectedFailure(
-    InlineAndInstallStrictExportTestExport.test_from_node_metadata_export_inline_and_install_strict  # noqa: F821
-)
-unittest.expectedFailure(
-    InlineAndInstallStrictExportTestExport.test_module_inline_and_install_strict  # noqa: F821
-)
-unittest.expectedFailure(
-    InlineAndInstallStrictExportTestExport.test_module_with_dict_container_inp_out_inline_and_install_strict  # noqa: F821
-)
-unittest.expectedFailure(
-    InlineAndInstallStrictExportTestExport.test_retrace_pre_autograd_inline_and_install_strict  # noqa: F821
-)
 
 if __name__ == "__main__":
     from torch._dynamo.test_case import run_tests

torch/_dynamo/trace_rules.py

@@ -3410,6 +3410,7 @@ MOD_INLINELIST = [
     "torch._dynamo.comptime",
     "torch._dynamo.polyfills",
     "torch._dynamo.test_case",
+    "torch._export.non_strict_utils",
     "torch._functorch._aot_autograd.subclass_parametrization",
     "torch._functorch.autograd_function",
     "torch._functorch.eager_transforms",
@@ -3430,6 +3431,7 @@ MOD_INLINELIST = [
     "torch.cuda.amp.autocast_mode",
     "torch.distributions",
     "torch.export._tree_utils",
+    "torch.export._unlift",
     "torch.export._wrapper_utils",
     "torch.fx._pytree",
     "torch.fx._symbolic_trace",