Merge pull request #2339 from ROCmSoftwarePlatform/hiprand-deps

Add rocm-ml-sdk package to tf-build image
ROCm · Dec 21, 2023 · e0ade0d · e0ade0d
2 parents 795c848 + dd1a014
commit e0ade0d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.rocm.cs7.txt b/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.rocm.cs7.txt
@@ -51,6 +51,7 @@ kernel-devel-uname-r
 # rocm packagaes
 libdrm-amdgpu
 rocm-dev
+rocm-ml-sdk
 miopen-hip 
 miopen-hip-devel 
 rocblas 

diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.rocm.txt b/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.rocm.txt
@@ -1,6 +1,7 @@
 # All required ROCM packages
 rocm-libs
 rocm-dev
+rocm-ml-sdk
 miopen-hip 
 miopen-hip-dev
 rocblas