From cf05a87e31ab772545eb2fc38612a75a1beec475 Mon Sep 17 00:00:00 2001
From: Sunita Nadampalli <nadampal@amazon.com>
Date: Wed, 6 Sep 2023 21:59:15 +0000
Subject: [PATCH] [linux-arm64] set "mkl_aarch64" bazel config for linux-arm64
 platform

---
 tensorflow-core/tensorflow-core-api/build.sh | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tensorflow-core/tensorflow-core-api/build.sh b/tensorflow-core/tensorflow-core-api/build.sh
index bf0713adf40..0119488c256 100755
--- a/tensorflow-core/tensorflow-core-api/build.sh
+++ b/tensorflow-core/tensorflow-core-api/build.sh
@@ -22,6 +22,11 @@ if [[ "${PLATFORM:-}" == macosx-arm64 ]]; then
   BUILD_FLAGS="$BUILD_FLAGS --config=macos_arm64"
 fi
 
+# Add platform specific flags
+if [[ "${PLATFORM:-}" == linux-arm64 ]]; then
+  BUILD_FLAGS="--config=mkl_aarch64"
+fi
+
 if [[ "${EXTENSION:-}" == *mkl* ]]; then
     BUILD_FLAGS="$BUILD_FLAGS --config=mkl"
 fi