File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
examples/models/llama2/custom_ops Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -58,7 +58,7 @@ cmake_install_executorch_libraries() {
58
58
fi
59
59
retry cmake -DBUCK2=" $BUCK " \
60
60
-DCMAKE_INSTALL_PREFIX=cmake-out \
61
- -DCMAKE_PREFIX_PATH=$( $PYTHON_EXECUTABLE -c " from distutils.sysconfig import get_python_lib; get_python_lib()" )
61
+ -DCMAKE_PREFIX_PATH=$( $PYTHON_EXECUTABLE -c " from distutils.sysconfig import get_python_lib; get_python_lib()" ) \
62
62
-DCMAKE_BUILD_TYPE=Release \
63
63
-DEXECUTORCH_BUILD_CUSTOM=ON \
64
64
-DEXECUTORCH_BUILD_EXTENSION_MODULE=ON \
Original file line number Diff line number Diff line change 21
21
if "CMAKE_INSTALL_PREFIX" in os .environ :
22
22
prefix = Path (os .environ ["CMAKE_INSTALL_PREFIX" ])
23
23
else :
24
- prefix = Path (__file__ ).parent .parent .parent .parent .parent .resolve () / "cmake-out"
24
+ prefix = (
25
+ Path (__file__ ).parent .parent .parent .parent .parent .resolve () / "cmake-out"
26
+ )
25
27
lib_path = prefix / "lib" / "libcustom_ops_aot_lib.so"
26
28
assert lib_path .exists (), f"{ lib_path } does not exist, please run cmake --install"
27
29
torch .ops .load_library (lib_path )
You can’t perform that action at this time.
0 commit comments