Skip to content

Update TensorFlow Versions and Patches #805

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
MattToast wants to merge 2 commits into
base: develop
Choose a base branch
from
Open

Update TensorFlow Versions and Patches #805

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ddf928e
Update TensorFlow for Numpy 2.0
MattToast May 14, 2026
1dd958e
Add changelog
MattToast May 15, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 6 additions & 4 deletions doc/changelog.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ Description
classes, and SmartDashboard integration
- Update copyright headers from 2021-2024 to 2021-2025 across the entire codebase
- Python 3.12 is now supported; where available, installed TensorFlow version
is now 2.16.2, PyTorch is 2.7.1.
is now 2.18.1, PyTorch is 2.7.1.
- Drop Python 3.9 support
- Add Numpy 2.0 support
- Terminate LSF and LSB support
Expand Down Expand Up @@ -76,10 +76,12 @@ Detailed Notes
tests, Docker files, shell scripts, and other supporting files to reflect the
new year.
([SmartSim-PR790](https://github.com/CrayLabs/SmartSim/pull/790))
- Python 3.12 is now supported. TensorFlow 2.16.2 and PyTorch 2.7.1 library
files are installed as part of `smart build` process when available. On Mac,
ONNX runtime 1.22.0 is now installed, together with ONNX 1.16.
- Python 3.12 is now supported. TensorFlow 2.18.1 (except on x86 MacOS where
the highest suported version is 2.16.2) and PyTorch 2.7.1 library files are
installed as part of `smart build` process when available. On Mac, ONNX
runtime 1.22.0 is now installed, together with ONNX 1.16.
([SmartSim-PR785](https://github.com/CrayLabs/SmartSim/pull/785))
([SmartSim-PR805](https://github.com/CrayLabs/SmartSim/pull/805))
- Python 3.9 will not be supported anymore, the last stable version of SmartSim
with support for Python 3.9 will be 0.8.
([SmartSim-PR781](https://github.com/CrayLabs/SmartSim/pull/781))
Expand Down
22 changes: 12 additions & 10 deletions smartsim/_core/_install/configs/mlpackages/DarwinARM64CPU.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,18 @@
"source_file": "src/backends/libtorch_c/CMakeLists.txt",
"regex": "set_property\\(TARGET\\storch_c\\sPROPERTY\\sCXX_STANDARD\\s(98|11|14)\\)",
"replacement": "set_property(TARGET torch_c PROPERTY CXX_STANDARD 17)"
},
}
]
},
{
"name": "libtensorflow",
"version": "2.18.1",
"pip_index": "",
"python_packages": [
"tensorflow==2.18.1"
],
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.18.1/libtensorflow-cpu-darwin-arm64.tar.gz",
"rai_patches": [
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
Expand All @@ -43,15 +54,6 @@
}
]
},
{
"name": "libtensorflow",
"version": "2.18.0",
"pip_index": "",
"python_packages": [
"tensorflow==2.18.1"
],
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.18.0/libtensorflow-cpu-darwin-arm64.tar.gz"
},
{
"name": "onnxruntime",
"version": "1.22.0",
Expand Down
22 changes: 12 additions & 10 deletions smartsim/_core/_install/configs/mlpackages/DarwinX64CPU.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,18 @@
"source_file": "src/backends/libtorch_c/CMakeLists.txt",
"regex": "set_property\\(TARGET\\storch_c\\sPROPERTY\\sCXX_STANDARD\\s(98|11|14)\\)",
"replacement": "set_property(TARGET torch_c PROPERTY CXX_STANDARD 17)"
},
}
]
},
{
"name": "libtensorflow",
"version": "2.16.2",
"pip_index": "",
"python_packages": [
"tensorflow==2.16.2"
],
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.16.2/libtensorflow-cpu-darwin-x86_64.tar.gz",
"rai_patches": [
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
Expand All @@ -43,15 +54,6 @@
}
]
},
{
"name": "libtensorflow",
"version": "2.16.2",
"pip_index": "",
"python_packages": [
"tensorflow==2.16.2"
],
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.16.2/libtensorflow-cpu-darwin-x86_64.tar.gz"
},
{
"name": "onnxruntime",
"version": "1.22.0",
Expand Down
22 changes: 12 additions & 10 deletions smartsim/_core/_install/configs/mlpackages/LinuxX64CPU.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,18 @@
"source_file": "src/backends/libtorch_c/CMakeLists.txt",
"regex": "set_property\\(TARGET\\storch_c\\sPROPERTY\\sCXX_STANDARD\\s(98|11|14)\\)",
"replacement": "set_property(TARGET torch_c PROPERTY CXX_STANDARD 17)"
},
}
]
},
{
"name": "libtensorflow",
"version": "2.18.1",
"pip_index": "",
"python_packages": [
"tensorflow==2.18.1"
],
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.18.1/libtensorflow-cpu-linux-x86_64.tar.gz",
"rai_patches": [
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
Expand All @@ -43,15 +54,6 @@
}
]
},
{
"name": "libtensorflow",
"version": "2.16.2",
"pip_index": "",
"python_packages": [
"tensorflow==2.16.2"
],
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.16.2/libtensorflow-cpu-linux-x86_64.tar.gz"
},
{
"name": "onnxruntime",
"version": "1.17.3",
Expand Down
22 changes: 12 additions & 10 deletions smartsim/_core/_install/configs/mlpackages/LinuxX64CUDA11.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,18 @@
"source_file": "src/backends/libtorch_c/CMakeLists.txt",
"regex": "set_property\\(TARGET\\storch_c\\sPROPERTY\\sCXX_STANDARD\\s(98|11|14)\\)",
"replacement": "set_property(TARGET torch_c PROPERTY CXX_STANDARD 17)"
},
}
]
},
{
"name": "libtensorflow",
"version": "2.14.1",
"pip_index": "",
"python_packages": [
"tensorflow==2.18.1"
],
"lib_source": "https://github.com/CrayLabs/ml_lib_builder/releases/download/v0.2/libtensorflow-2.14.1-linux-x64-cuda-11.8.0.tgz",
"rai_patches":[
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
Expand All @@ -43,15 +54,6 @@
}
]
},
{
"name": "libtensorflow",
"version": "2.14.1",
"pip_index": "",
"python_packages": [
"tensorflow==2.14.1"
],
"lib_source": "https://github.com/CrayLabs/ml_lib_builder/releases/download/v0.2/libtensorflow-2.14.1-linux-x64-cuda-11.8.0.tgz"
},
{
"name": "onnxruntime",
"version": "1.17.3",
Expand Down
30 changes: 15 additions & 15 deletions smartsim/_core/_install/configs/mlpackages/LinuxX64CUDA12.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,35 +28,35 @@
"source_file": "src/backends/libtorch_c/CMakeLists.txt",
"regex": "set_property\\(TARGET\\storch_c\\sPROPERTY\\sCXX_STANDARD\\s(98|11|14)\\)",
"replacement": "set_property(TARGET torch_c PROPERTY CXX_STANDARD 17)"
},
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
"regex": "TF_Input inputs",
"replacement": "TF_Output inputs"
},
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
"regex": "TF_Input port",
"replacement": "TF_Output port"
}
]
},
{
"name": "libtensorflow",
"version": "2.16.2",
"version": "2.18.1",
"pip_index": "",
"python_packages": [
"tensorflow==2.16.2"
"tensorflow==2.18.1"
],
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.16.2/libtensorflow-gpu-linux-x86_64.tar.gz",
"lib_source": "https://storage.googleapis.com/tensorflow/versions/2.18.1/libtensorflow-gpu-linux-x86_64.tar.gz",
"rai_patches": [
{
"description": "Patch RedisAI to point to correct tsl directory",
"source_file": "CMakeLists.txt",
"regex": "INCLUDE_DIRECTORIES\\(\\$\\{depsAbs\\}/libtensorflow/include\\)",
"replacement": "INCLUDE_DIRECTORIES(${depsAbs}/libtensorflow/include ${depsAbs}/libtensorflow/include/external/local_tsl)"
},
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
"regex": "TF_Input inputs",
"replacement": "TF_Output inputs"
},
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
"regex": "TF_Input port",
"replacement": "TF_Output port"
}
]
},
Expand Down
12 changes: 0 additions & 12 deletions smartsim/_core/_install/configs/mlpackages/LinuxX64ROCM6.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,18 +40,6 @@
"source_file": "../package/libtorch/share/cmake/Caffe2/Caffe2Targets.cmake",
"regex": "/opt/rocm",
"replacement": "$ENV{ROCM_PATH}"
},
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
"regex": "TF_Input inputs",
"replacement": "TF_Output inputs"
},
{
"description": "Fix the type in a Tensorflow function signature",
"source_file": "src/backends/tensorflow.c",
"regex": "TF_Input port",
"replacement": "TF_Output port"
}
]
}
Expand Down
Loading