From bd42ce07634a7044b5166aa9ce1832d4f7a42720 Mon Sep 17 00:00:00 2001
From: Tongze Wang <wtongze@gmail.com>
Date: Wed, 11 Sep 2024 19:58:43 -0700
Subject: [PATCH] fix: tf gfx check

---
 tensorflow.dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow.dockerfile b/tensorflow.dockerfile
index ffcbf7f..13c6e81 100644
--- a/tensorflow.dockerfile
+++ b/tensorflow.dockerfile
@@ -47,6 +47,8 @@ RUN python${PYTHON_VERSION} -m pip install -U --user keras_preprocessing --no-de
 RUN sed -i -e "s/python3/python${PYTHON_VERSION}/g" -e "s/pip3/python${PYTHON_VERSION} -m pip/g" build_rocm_python3
 RUN sudo ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
 
+RUN sed -i 's/"gfx900",/"gfx803", "gfx900",/g' tensorflow/compiler/xla/stream_executor/device_description.h
+
 ENV ROCM_PATH=/opt/rocm-${ROCM_VERSION}
 ENV ROCM_TOOLKIT_PATH=${ROCM_PATH}
 RUN ./build_rocm_python3 ${ROCM_PATH}