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

[Codemod][AddExplicitStrictExportForTrainingInferenceArg] caffe2/torch/ao (#150826)

Browse Source 
Differential Revision: D72615631

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150826
Approved by: https://github.com/ydwu4
This commit is contained in:
Yanan Cao (PyTorch) 2025-04-08 18:49:22 +00:00 committed by PyTorch MergeBot
parent aafc4b6188
commit c36d9b0d8d
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/ao/quantization/pt2e/lowering.py
View File

@ -50,7 +50,7 @@ def lower_pt2e_quantized_to_x86(
m.recompile()
lowered_model = (
torch.export.export_for_training(model, example_inputs)
torch.export.export_for_training(model, example_inputs, strict=True)
.run_decompositions(_post_autograd_decomp_table())
.module()
)