Skip to content

Commit e111fdf

Browse files
guangy10Guang Yang
authored and
Guang Yang
committed
[RELEASE-ONLY CHANGES] Update release/0.2 to pin and test against PyTorch release/2.3 (#2818)
1 parent a681e56 commit e111fdf

File tree

4 files changed

+9
-14
lines changed

4 files changed

+9
-14
lines changed

.ci/docker/ci_commit_pins/pytorch.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0a038cf0cff2d071b7359ac0491fd2ba7798a438
1+
release/2.3

backends/arm/arm_quantizer_utils.py

+2-2
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)

install_requirements.sh

+5-10
Original file line numberDiff line numberDiff line change
@@ -49,19 +49,14 @@ done
4949
# Since ExecuTorch often uses main-branch features of pytorch, only the nightly
5050
# pip versions will have the required features. The NIGHTLY_VERSION value should
5151
# agree with the third-party/pytorch pinned submodule commit.
52-
#
53-
# NOTE: If a newly-fetched version of the executorch repo changes the value of
54-
# NIGHTLY_VERSION, you should re-run this script to install the necessary
55-
# package versions.
56-
NIGHTLY_VERSION=dev20240324
5752

5853
# The pip repository that hosts nightly torch packages.
59-
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cpu"
54+
TORCH_URL="https://download.pytorch.org/whl/test/cpu"
6055

6156
# pip packages needed by exir.
6257
EXIR_REQUIREMENTS=(
63-
torch=="2.4.0.${NIGHTLY_VERSION}"
64-
torchvision=="0.19.0.${NIGHTLY_VERSION}" # For testing.
58+
torch=="2.3.0"
59+
torchvision=="0.18.0"
6560
)
6661

6762
# pip packages needed for development.
@@ -77,7 +72,7 @@ DEVEL_REQUIREMENTS=(
7772
# TODO(dbort): Make each example publish its own requirements.txt
7873
EXAMPLES_REQUIREMENTS=(
7974
timm==0.6.13
80-
torchaudio=="2.2.0.${NIGHTLY_VERSION}"
75+
torchaudio=="2.3.0"
8176
torchsr==1.0.4
8277
transformers==4.38.2
8378
)
@@ -92,7 +87,7 @@ REQUIREMENTS_TO_INSTALL=(
9287

9388
# Install the requirements. `--extra-index-url` tells pip to look for package
9489
# versions on the provided URL if they aren't available on the default URL.
95-
pip install --extra-index-url "${TORCH_NIGHTLY_URL}" \
90+
pip install --extra-index-url "${TORCH_URL}" \
9691
"${REQUIREMENTS_TO_INSTALL[@]}"
9792

9893
#

third-party/pytorch

Submodule pytorch updated 1362 files

0 commit comments

Comments
 (0)