OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Revert "[export] Copy gm before calling PassManager" for test or build failures (#108441)

Browse Source 
Test Plan: CI

Differential Revision: D48916322

Pull Request resolved: https://github.com/pytorch/pytorch/pull/108441
Approved by: https://github.com/cccclai
This commit is contained in:
Digant Desai 2023-09-05 19:21:01 +00:00 committed by PyTorch MergeBot
parent 0b44fdfaec
commit a9a6423261
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/export/__init__.py
View File

@ -399,7 +399,7 @@ class ExportedProgram:
)
pm = PassManager(list(passes))
res = pm(copy.deepcopy(self.graph_module))
res = pm(self.graph_module)
transformed_gm = res.graph_module if res is not None else self.graph_module
assert transformed_gm is not None