diff options
Diffstat (limited to 'compiler/one-cmds/CMakeLists.txt')
-rw-r--r-- | compiler/one-cmds/CMakeLists.txt | 21 |
1 files changed, 14 insertions, 7 deletions
diff --git a/compiler/one-cmds/CMakeLists.txt b/compiler/one-cmds/CMakeLists.txt index 7772b5340..917bbdaf6 100644 --- a/compiler/one-cmds/CMakeLists.txt +++ b/compiler/one-cmds/CMakeLists.txt @@ -36,13 +36,19 @@ set(ONE_COMMAND_FILES ) # TODO find better way for per-platform files -if(ONE_UBUNTU_CODENAME_JAMMY) - # NOTE copy one-prepare-venv.u2204 as build/../one-prepare-venv +if(CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64") + # NOTE copy one-prepare-venv.aarch64 as build/../one-prepare-venv # and install build/../one-prepare-venv file - list(APPEND ONE_COMMAND_FILES one-prepare-venv.u2204) -else() - list(APPEND ONE_COMMAND_FILES one-prepare-venv) -endif() + list(APPEND ONE_COMMAND_FILES one-prepare-venv.aarch64) +else(CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64") + if(ONE_UBUNTU_CODENAME_BIONIC) + # NOTE copy one-prepare-venv.u1804 as build/../one-prepare-venv + # and install build/../one-prepare-venv file + list(APPEND ONE_COMMAND_FILES one-prepare-venv.u1804) + else(ONE_UBUNTU_CODENAME_BIONIC) + list(APPEND ONE_COMMAND_FILES one-prepare-venv) + endif(ONE_UBUNTU_CODENAME_BIONIC) +endif(CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64") # pytorch importer is an experimental feature, it is not used in default configuration if(ENABLE_ONE_IMPORT_PYTORCH) @@ -113,7 +119,8 @@ install(FILES ${MODEL2NNPKG} RENAME "model2nnpkg") # make python directory -set(ONE_PYTHON_FILES constant.py +set(ONE_PYTHON_FILES backends.py + constant.py export_constant.py make_cmd.py CfgRunner.py |