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

[dynamo] Reland #104317 - Lazy disable_dynamo API out-of-dynamo (#104664)

Browse Source 
Internal failed because of torch.deploy issues with disable_dynamo in fx/* and _jit/* files. Removing disable_dynamo for both. Added a comment in the code.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/104664
Approved by: https://github.com/wconstab
This commit is contained in:
Animesh Jain
2023-07-05 21:38:53 +00:00
committed by PyTorch MergeBot
parent d3589c9456
commit 0444f9f85b
7 changed files with 49 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/onnx/utils.py
View File

@ -1231,6 +1231,7 @@ def _model_to_graph(
@_beartype.beartype
@torch._disable_dynamo
def export_to_pretty_string(
model,
args,