Skip to content

Commit

Permalink
Default to use training IR (pytorch#137804)
Browse files Browse the repository at this point in the history
Summary: Since capture_pre_autograd_graph is deprecated and will be deleted soon, we default this option to true.

Test Plan: CI

Reviewed By: tugsbayasgalan

Differential Revision: D64254236

Pull Request resolved: pytorch#137804
Approved by: https://github.com/tugsbayasgalan
  • Loading branch information
yushangdi authored and pytorchmergebot committed Oct 11, 2024
1 parent c254901 commit 432c3fe
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions torch/ao/quantization/pt2e/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -363,8 +363,7 @@ def _get_aten_graph_module_for_pattern(
pattern: Callable,
example_inputs: Tuple[Any, ...],
is_cuda: bool = False,
*,
using_training_ir: bool,
using_training_ir: bool = True,
**kwargs,
) -> GraphModule:
"""
Expand Down

0 comments on commit 432c3fe

Please sign in to comment.