From dd1a01462b67436d7792801aa28e7af0d1a9c405 Mon Sep 17 00:00:00 2001 From: Jason Furmanek Date: Wed, 4 Oct 2023 14:42:19 +0000 Subject: [PATCH] Add rocm-ml-sdk package to tf-build image --- .../tools/tf_sig_build_dockerfiles/devel.packages.rocm.cs7.txt | 1 + .../tools/tf_sig_build_dockerfiles/devel.packages.rocm.txt | 1 + 2 files changed, 2 insertions(+) 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 index d8379275b9fc7e..51b977fc946ad0 100644 --- 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 index a616b90bac0dbf..a5fb84db156842 100644 --- 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