From f269324c13cbd9529237d2cbaee6e88c40f3a603 Mon Sep 17 00:00:00 2001 From: Jason Furmanek Date: Wed, 10 Jul 2024 17:36:34 +0000 Subject: [PATCH] Update build script --- build_rocm_python3 | 18 +++++------------- 1 file changed, 5 insertions(+), 13 deletions(-) diff --git a/build_rocm_python3 b/build_rocm_python3 index 82465d3aac2a76..9b711c65da65a6 100755 --- a/build_rocm_python3 +++ b/build_rocm_python3 @@ -25,12 +25,8 @@ while getopts "hrn" opt; do done shift "$((OPTIND-1))" -# This is not a release branch, so force a nightly build -# TODO remove this when branching for release -nightly=true - # First positional argument (if any) specifies the ROCM_INSTALL_DIR -ROCM_INSTALL_DIR=/opt/rocm-6.1.1 +ROCM_INSTALL_DIR=/opt/rocm-6.2.0 if [[ -n $1 ]]; then ROCM_INSTALL_DIR=$1 fi @@ -55,15 +51,13 @@ if [ -f /usertools/rocm.bazelrc ]; then rm -f $TF_PKG_LOC/tf_nightly_rocm*.whl export project_name=tf_nightly_rocm python3 tensorflow/tools/ci_build/update_version.py --nightly --rocm_version && - bazel --bazelrc=/usertools/rocm.bazelrc build $RESOURCE_OPTION --config=rocm --repo_env=WHEEL_NAME=tf_nightly_rocm --action_env=project_name=tf_nightly_rocm --action_env=TF_PYTHON_VERSION=$PYTHON_VERSION tensorflow/tools/pip_package:wheel --verbose_failures && - cp ./bazel-bin/tensorflow/tools/pip_package/wheel_house/tf_nightly_rocm*.whl $TF_PKG_LOC/ && + bazel --bazelrc=/usertools/rocm.bazelrc build $RESOURCE_OPTION --config=rocm --repo_env=WHEEL_NAME=tf_nightly_rocm --repo_env=OUTPUT_PATH=$TF_PKG_LOC --action_env=project_name=tf_nightly_rocm --action_env=TF_PYTHON_VERSION=$PYTHON_VERSION tensorflow/tools/pip_package:wheel --verbose_failures && pip3 install --upgrade $TF_PKG_LOC/tf_nightly_rocm*.whl else # Remove any previous builds and build release rm -f $TF_PKG_LOC/tensorflow*.whl python3 tensorflow/tools/ci_build/update_version.py --rocm_version && - bazel --bazelrc=/usertools/rocm.bazelrc build $RESOURCE_OPTION --config=rocm --repo_env=WHEEL_NAME=tensorflow_rocm --action_env=project_name=tensorflow_rocm --action_env=TF_PYTHON_VERSION=$PYTHON_VERSION tensorflow/tools/pip_package:wheel --verbose_failures && - cp ./bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow*.whl $TF_PKG_LOC/ && + bazel --bazelrc=/usertools/rocm.bazelrc build $RESOURCE_OPTION --config=rocm --repo_env=WHEEL_NAME=tensorflow_rocm --repo_env=OUTPUT_PATH=$TF_PKG_LOC --action_env=project_name=tensorflow_rocm --action_env=TF_PYTHON_VERSION=$PYTHON_VERSION tensorflow/tools/pip_package:wheel --verbose_failures && pip3 install --upgrade $TF_PKG_LOC/tensorflow*.whl fi else @@ -73,14 +67,12 @@ else if [[ -n $nightly ]]; then # Remove any previous builds and build nightly rm -f $TF_PKG_LOC/tf_nightly_rocm*.whl - bazel build $RESOURCE_OPTION --config=opt --config=rocm --repo_env=WHEEL_NAME=tf_nightly_rocm --action_env=project_name=tf_nightly_rocm //tensorflow/tools/pip_package:wheel --verbose_failures && - cp ./bazel-bin/tensorflow/tools/pip_package/wheel_house/tf_nightly_rocm*.whl $TF_PKG_LOC/ && + bazel build $RESOURCE_OPTION --config=opt --config=rocm --repo_env=WHEEL_NAME=tf_nightly_rocm --repo_env=OUTPUT_PATH=$TF_PKG_LOC --action_env=project_name=tf_nightly_rocm //tensorflow/tools/pip_package:wheel --verbose_failures && pip3 install --upgrade $TF_PKG_LOC/tf_nightly_rocm*.whl else # Remove any previous builds and build release rm -f $TF_PKG_LOC/tensorflow*.whl - bazel build $RESOURCE_OPTION --config=opt --config=rocm --repo_env=WHEEL_NAME=tensorflow_rocm --action_env=project_name=tensorflow_rocm //tensorflow/tools/pip_package:wheel --verbose_failures && - cp ./bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow*.whl $TF_PKG_LOC/ && + bazel build $RESOURCE_OPTION --config=opt --config=rocm --repo_env=WHEEL_NAME=tensorflow_rocm --repo_env=OUTPUT_PATH=$TF_PKG_LOC --action_env=project_name=tensorflow_rocm //tensorflow/tools/pip_package:wheel --verbose_failures && pip3 install --upgrade $TF_PKG_LOC/tensorflow*.whl fi fi