Skip to content

Commit 051387e

Browse files
authored
Update install_executorch.py
1 parent 6f84f98 commit 051387e

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

install_executorch.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,11 @@
1717
from contextlib import contextmanager
1818
from typing import List, Tuple
1919

20-
from install_requirements import install_requirements, python_is_compatible, TORCH_URL
20+
from install_requirements import (
21+
install_requirements,
22+
python_is_compatible,
23+
TORCH_NIGHTLY_URL,
24+
)
2125

2226
# Set up logging
2327
logging.basicConfig(
@@ -49,7 +53,7 @@ def clean():
4953

5054

5155
# Please keep this insync with `ShouldBuild.pybindings` in setup.py.
52-
VALID_PYBINDS = ["coreml", "mps", "xnnpack", "training"]
56+
VALID_PYBINDS = ["coreml", "mps", "xnnpack", "training", "openvino"]
5357

5458

5559
################################################################################
@@ -198,14 +202,10 @@ def main(args):
198202
use_pytorch_nightly = True
199203

200204
wants_pybindings_off, pybind_defines = _list_pybind_defines(args)
201-
if not wants_pybindings_off:
202-
if len(pybind_defines) > 0:
203-
# If the user explicitly provides a list of bindings, just use them
204-
cmake_args += pybind_defines
205-
else:
206-
# If the user has not set pybindings off but also has not provided
207-
# a list, then turn on xnnpack by default
208-
cmake_args.append("-DEXECUTORCH_BUILD_XNNPACK=ON")
205+
if wants_pybindings_off:
206+
cmake_args.append("-DEXECUTORCH_BUILD_PYBIND=OFF")
207+
else:
208+
cmake_args += pybind_defines
209209

210210
if args.clean:
211211
clean()
@@ -251,7 +251,7 @@ def main(args):
251251
"--no-build-isolation",
252252
"-v",
253253
"--extra-index-url",
254-
TORCH_URL,
254+
TORCH_NIGHTLY_URL,
255255
],
256256
check=True,
257257
)

0 commit comments

Comments
 (0)