frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[dynamo][export] Add some missing trace rules (#164080)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/164080
Approved by: https://github.com/tugsbayasgalan
This commit is contained in:
Animesh Jain
2025-09-28 21:46:28 -07:00
committed by PyTorch MergeBot
parent 5ddad22196
commit e1e5e040cd
2 changed files with 2 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/export/test_export_with_inline_and_install.py
View File

@ -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

2
torch/_dynamo/trace_rules.py
View File

@ -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",