Skip to content

Commit f8e469d

Browse files
jathuBujSet
authored andcommitted
Enable training for pybindings (pytorch#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 7e13bcd commit f8e469d

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)