Skip to content

Commit 286799c

Browse files
swolchokfacebook-github-bot
authored andcommitted
Include optimized kernels in pybindings' portable_lib if building them (pytorch#5520)
Summary: Pull Request resolved: pytorch#5520 reserved ghstack-source-id: 243867516 exported-using-ghexport Reviewed By: larryliu0820 Differential Revision: D63147278 fbshipit-source-id: d5aefbf2509a1eca4c32bbbe7224e7b996fa1e57
1 parent 28c2ab6 commit 286799c

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

CMakeLists.txt

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -680,11 +680,16 @@ if(EXECUTORCH_BUILD_PYBIND)
680680
etdump
681681
executorch
682682
extension_data_loader
683-
portable_ops_lib
684683
util
685684
torch
686685
)
687686

687+
if(EXECUTORCH_BUILD_KERNELS_OPTIMIZED)
688+
list(APPEND _dep_libs optimized_native_cpu_ops_lib)
689+
else()
690+
list(APPEND _dep_libs portable_ops_lib)
691+
endif()
692+
688693
if(EXECUTORCH_BUILD_COREML)
689694
list(APPEND _dep_libs coremldelegate)
690695
endif()

0 commit comments

Comments
 (0)