Fix build breaks due to

i. migrating third_party/mkl to @local_xla//xla/tsl/mkl
ii. removal of cuda.h header
iii. replace build_pip_package rule with wheel in build_rocm_python3 script

Author: hsharsha

build_rocm_python3

@@ -20,7 +20,7 @@ while getopts "hrn" opt; do
         restriction=true
         ;;
     n)
-	nightly=true
+    nightly=true
     esac
 done
 shift "$((OPTIND-1))"
@@ -47,35 +47,22 @@ else
 fi
 
 if [ -f /usertools/rocm.bazelrc ]; then
-	# Use the bazelrc files in /usertools if available
-    	TF_PKG_LOC=bazel-bin/tensorflow/tools/pip_package/wheel_house
-	if [[ -n $nightly ]]; then
-		# Remove any previous builds and build nightly
-	        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 &&
-		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 &&
-		pip3 install --upgrade $TF_PKG_LOC/tensorflow*.whl
-	fi
+    # Use the bazelrc files in /usertools if available
+    # Also, this is likely a tensorflow-build container so put the whl in /tf/pkg
+    TF_PKG_LOC=/tf/pkg
+    # Remove any previous builds and build nightly
+    rm -f $TF_PKG_LOC/tensorflow*.whl
+    python3 tensorflow/tools/ci_build/update_version.py --nightly --rocm_version &&
+    bazel --bazelrc=/usertools/rocm.bazelrc build $RESOURCE_OPTION --config=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/ &&
+    pip3 install --upgrade $TF_PKG_LOC/tensorflow*.whl
 else
-	# Legacy style: run configure then build
-        TF_PKG_LOC=bazel-bin/tensorflow/tools/pip_package/wheel_house
-	yes "" | TF_NEED_CLANG=0 ROCM_PATH=$ROCM_INSTALL_DIR TF_NEED_ROCM=1 PYTHON_BIN_PATH=/usr/bin/python3 ./configure &&
-	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 &&
-		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 &&
-		pip3 install --upgrade $TF_PKG_LOC/tensorflow*.whl
-	fi
+    # Legacy style: run configure then build
+    TF_PKG_LOC=/tmp/tensorflow_pkg
+    yes "" | TF_NEED_CLANG=0 ROCM_PATH=$ROCM_INSTALL_DIR TF_NEED_ROCM=1 PYTHON_BIN_PATH=/usr/bin/python3 ./configure &&
+    # Remove any previous builds and build nightly
+    rm -f $TF_PKG_LOC/tensorflow*.whl
+    bazel build $RESOURCE_OPTION --config=opt --config=rocm //tensorflow/tools/pip_package:wheel --verbose_failures &&
+    cp ./bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow*.whl $TF_PKG_LOC/ &&
+    pip3 install --upgrade $TF_PKG_LOC/tf_nightly_rocm*.whl
 fi