Skip to content

Commit d66d17d

Browse files
committed
Explicitly Add 'DEXECUTORCH_BUILD_EXTENSION_TENSOR' arg when 'xnnpack' is enabled
1 parent 360c9bb commit d66d17d

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
@@ -191,6 +191,8 @@ def _list_pybind_defines(args) -> Tuple[bool, List[str]]:
191191
cmake_args.append("-DEXECUTORCH_BUILD_EXTENSION_TRAINING=ON")
192192
else:
193193
cmake_args.append(f"-DEXECUTORCH_BUILD_{pybind_arg.upper()}=ON")
194+
if pybind_arg == "xnnpack":
195+
cmake_args.append("-DEXECUTORCH_BUILD_EXTENSION_TENSOR=ON")
194196

195197
return False, cmake_args
196198

0 commit comments

Comments
 (0)