fix: tf gfx check

2024-09-11 19:58:43 -07:00
parent 7bd4c10b51
commit bd42ce0763
1 changed files with 2 additions and 0 deletions
@@ -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}