Skip to content

Commit 955c863

Browse files
authored
Enable training for pybindings (#12144)
### Summary Let's support the (experimental) training modules out of the box for our wheels. ### Test plan ``` $ ./install_executorch.sh $ python3 >>> import torch >>> from executorch.extension.training.pybindings import _training_lib ``` cc @JacobSzwejbka
1 parent 3ba0466 commit 955c863

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tools/cmake/preset/pybind.cmake

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ set_overridable_option(EXECUTORCH_BUILD_KERNELS_CUSTOM_AOT ON)
2020
set_overridable_option(EXECUTORCH_BUILD_EXTENSION_FLAT_TENSOR ON)
2121
set_overridable_option(EXECUTORCH_BUILD_EXTENSION_DATA_LOADER ON)
2222
set_overridable_option(EXECUTORCH_BUILD_KERNELS_OPTIMIZED ON)
23+
set_overridable_option(EXECUTORCH_BUILD_EXTENSION_MODULE ON)
24+
set_overridable_option(EXECUTORCH_BUILD_EXTENSION_TRAINING ON)
2325

2426

2527
if(CMAKE_SYSTEM_NAME STREQUAL "Darwin")

0 commit comments

Comments
 (0)