File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -139,25 +139,12 @@ run_core_tests () {
139139 echo " === [$LABEL ] Installing wheel & deps ==="
140140 " $PIPBIN " install --upgrade pip
141141 " $PIPBIN " install " $WHEEL_FILE "
142- TORCH_VERSION=$(
143- " $PYBIN " - << 'PY '
144- import runpy
145- module_vars = runpy.run_path("torch_pin.py")
146- print(module_vars["TORCH_VERSION"])
147- PY
148- )
142+ TORCH_VERSION=2.10.0
149143
150- NIGHTLY_VERSION=$(
151- " $PYBIN " - << 'PY '
152- import runpy
153- module_vars = runpy.run_path("torch_pin.py")
154- print(module_vars["NIGHTLY_VERSION"])
155- PY
156- )
157- echo " === [$LABEL ] Install torch==${TORCH_VERSION} .${NIGHTLY_VERSION} ==="
144+ echo " === [$LABEL ] Install torch==${TORCH_VERSION} ==="
158145
159146 # Install torchao based on the pinned PyTorch version
160- " $PIPBIN " install torch==" ${TORCH_VERSION} . ${NIGHTLY_VERSION} " --index-url " https://download.pytorch.org/whl/nightly/cpu "
147+ " $PIPBIN " install torch==" ${TORCH_VERSION} "
161148 " $PIPBIN " install wheel
162149
163150 # Install torchao based on the pinned commit from third-party/ao submodule
You can’t perform that action at this time.
0 commit comments