Skip to content

Commit 0beadcc

Browse files
authored
Enable pybind with mps
Differential Revision: D69497059 Pull Request resolved: #8401
1 parent 6727fea commit 0beadcc

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

install_executorch.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,8 @@ def main(args):
165165
if pybind_arg == "training":
166166
CMAKE_ARGS += " -DEXECUTORCH_BUILD_EXTENSION_TRAINING=ON"
167167
os.environ["EXECUTORCH_BUILD_TRAINING"] = "ON"
168+
elif pybind_arg == "mps":
169+
CMAKE_ARGS += " -DEXECUTORCH_BUILD_MPS=ON"
168170
else:
169171
CMAKE_ARGS += f" -DEXECUTORCH_BUILD_{pybind_arg.upper()}=ON"
170172
EXECUTORCH_BUILD_PYBIND = "ON"

0 commit comments

Comments
 (0)