diff --git a/tensorflow/tools/ci_build/Dockerfile.rocm b/tensorflow/tools/ci_build/Dockerfile.rocm index 4ab2f62a43d..eb0d0b4db98 100644 --- a/tensorflow/tools/ci_build/Dockerfile.rocm +++ b/tensorflow/tools/ci_build/Dockerfile.rocm @@ -78,7 +78,7 @@ ENV PATH="$ROCM_PATH/bin:${PATH}" ENV PATH="$OPENCL_ROOT/bin:${PATH}" # Add target file to help determine which device(s) to build for -RUN bash -c 'echo -e "gfx803\ngfx900\ngfx906" >> ${ROCM_PATH}/bin/target.lst' +RUN bash -c 'echo -e "gfx803\ngfx900\ngfx906\ngfx908" >> ${ROCM_PATH}/bin/target.lst' # Need to explicitly create the $ROCM_PATH/.info/version file to workaround what seems to be a bazel bug # The env vars being set via --action_env in .bazelrc and .tf_configure.bazelrc files are sometimes