Skip to content

Commit 84e02c5

Browse files
committed
[RELEASE-ONLY CHANGES] Update release/0.2 to pin and test against PyTorch release/2.3
1 parent 4e17c7c commit 84e02c5

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.ci/docker/ci_commit_pins/pytorch.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
a52b4e22571507abc35c2d47de138497190d2e0a
1+
8194fae625835394ab088985e08aabe65926e524

backends/arm/arm_quantizer_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from torch.ao.quantization.pt2e.utils import (
2424
_conv1d_bn_example_inputs,
2525
_conv2d_bn_example_inputs,
26-
_get_aten_graph_module_for_pattern,
26+
get_aten_graph_module,
2727
)
2828
from torch.ao.quantization.quantizer import (
2929
QuantizationAnnotation,
@@ -478,7 +478,7 @@ def _do_annotate_conv_bn(
478478
# Match against all conv dimensions and cuda variants
479479
for (conv_fn, example_inputs), is_cuda, relu_is_inplace in combinations:
480480
pattern = _get_pattern(conv_fn, relu_is_inplace, has_relu)
481-
pattern = _get_aten_graph_module_for_pattern(pattern, example_inputs, is_cuda)
481+
pattern = get_aten_graph_module(pattern, example_inputs, is_cuda)
482482
pattern.graph.eliminate_dead_code()
483483
pattern.recompile()
484484
matcher = SubgraphMatcherWithNameNodeMap(pattern, ignore_literals=True)

0 commit comments

Comments
 (0)