@@ -32,6 +32,19 @@ else()
3232 return ()
3333 endif ()
3434
35+ # NOTE does not check for non python 3.9 or python 3.11
36+ if (Python_VERSION VERSION_LESS 3.8.99)
37+ set (PYTHON_VERSION_MINOR 8)
38+ elseif (Python_VERSION VERSION_LESS 3.10.99)
39+ set (PYTHON_VERSION_MINOR 10)
40+ elseif (Python_VERSION VERSION_LESS 3.12.99)
41+ set (PYTHON_VERSION_MINOR 12)
42+ else ()
43+ # TODO support higher version
44+ message (STATUS "Build common-artifacts: FAILED (Unsupported python: ${Python_VERSION} )" )
45+ return ()
46+ endif ()
47+
3548 set (PYTHON_EXECUTABLE ${Python_EXECUTABLE} )
3649endif ()
3750
@@ -59,18 +72,55 @@ if(DEFINED ENV{ONE_PIP_OPTION_TRUST_HOST})
5972endif ()
6073
6174set (PY_PKG_LIST )
62- if (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
63- # NOTE `tensorflow-cpu` package is not available for aarch64, so we use `tensorflow` package.
64- list (APPEND PY_PKG_LIST "tensorflow==2.12.1" )
65- else (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
66- list (APPEND PY_PKG_LIST "tensorflow-cpu==2.12.1" )
67- endif (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
68- list (APPEND PY_PKG_LIST "flatbuffers==23.5.26" )
69- list (APPEND PY_PKG_LIST "protobuf==4.23.3" )
70- list (APPEND PY_PKG_LIST "pydot==1.4.2" )
71- list (APPEND PY_PKG_LIST "pytest==7.4.3" )
72- list (APPEND PY_PKG_LIST "h5py==3.11.0" )
73- list (APPEND PY_PKG_LIST "cffi==1.16.0" )
75+ if (PYTHON_VERSION_MINOR EQUAL 8)
76+ # python3.8; Ubuntu 18.04, 20.04
77+ if (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
78+ # NOTE `tensorflow-cpu` package is not available for aarch64, so we use `tensorflow` package.
79+ list (APPEND PY_PKG_LIST "tensorflow==2.12.1" )
80+ else (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
81+ list (APPEND PY_PKG_LIST "tensorflow-cpu==2.12.1" )
82+ endif (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
83+ list (APPEND PY_PKG_LIST "flatbuffers==23.5.26" )
84+ list (APPEND PY_PKG_LIST "protobuf==4.23.3" )
85+ list (APPEND PY_PKG_LIST "pydot==1.4.2" )
86+ list (APPEND PY_PKG_LIST "pytest==7.4.3" )
87+ list (APPEND PY_PKG_LIST "h5py==3.11.0" )
88+ list (APPEND PY_PKG_LIST "cffi==1.16.0" )
89+ elseif (PYTHON_VERSION_MINOR EQUAL 10)
90+ # python 3.10; Ubuntu 22.04
91+ # TF2.19.0 release at 2025-03-12
92+ if (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
93+ # NOTE `tensorflow-cpu` package is not available for aarch64, so we use `tensorflow` package.
94+ list (APPEND PY_PKG_LIST "tensorflow==2.19.0" )
95+ else (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
96+ list (APPEND PY_PKG_LIST "tensorflow-cpu==2.19.0" )
97+ endif (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
98+ list (APPEND PY_PKG_LIST "flatbuffers==24.3.25" )
99+ list (APPEND PY_PKG_LIST "protobuf==4.23.3" )
100+ list (APPEND PY_PKG_LIST "pydot==1.4.2" )
101+ list (APPEND PY_PKG_LIST "pytest==7.4.3" )
102+ list (APPEND PY_PKG_LIST "h5py==3.11.0" )
103+ list (APPEND PY_PKG_LIST "cffi==1.16.0" )
104+ elseif (PYTHON_VERSION_MINOR EQUAL 12)
105+ # python 3.12; Ubuntu 24.04
106+ # TF2.19.0 release at 2025-03-12
107+ if (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
108+ # NOTE `tensorflow-cpu` package is not available for aarch64, so we use `tensorflow` package.
109+ list (APPEND PY_PKG_LIST "tensorflow==2.19.0" )
110+ else (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
111+ list (APPEND PY_PKG_LIST "tensorflow-cpu==2.19.0" )
112+ endif (CMAKE_HOST_SYSTEM_PROCESSOR STREQUAL "aarch64" )
113+ list (APPEND PY_PKG_LIST "flatbuffers==24.3.25" )
114+ list (APPEND PY_PKG_LIST "protobuf==4.23.3" )
115+ list (APPEND PY_PKG_LIST "pydot==1.4.2" )
116+ list (APPEND PY_PKG_LIST "pytest==7.4.3" )
117+ list (APPEND PY_PKG_LIST "h5py==3.11.0" )
118+ list (APPEND PY_PKG_LIST "cffi==1.16.0" )
119+ list (APPEND PY_PKG_LIST "numpy==1.26.4" )
120+ else ()
121+ # Should not enter here
122+ message (FATAL "Build common-artifacts: FAILED (Invalid pyhon version)" )
123+ endif ()
74124
75125add_custom_command (
76126 OUTPUT REQUIREMENTS_OVERLAY_PATH_clean
0 commit comments