From 0a35cbb2b32b1238565a224163cda8f5d571b407 Mon Sep 17 00:00:00 2001
From: Brian Yang <125406446+gpupuck@users.noreply.github.com>
Date: Thu, 12 Dec 2024 13:50:31 -0800
Subject: [PATCH] Disable MKL DNN build for ARM only due to JAX PR#23225

---
 .github/container/build-jax.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/container/build-jax.sh b/.github/container/build-jax.sh
index 95cf5246b..b179e9456 100755
--- a/.github/container/build-jax.sh
+++ b/.github/container/build-jax.sh
@@ -185,6 +185,8 @@ case "${CPU_ARCH}" in
         ;;
     "arm64")
         export CC_OPT_FLAGS="-march=armv8-a"
+        # ARM ACL build issue introduced in PR#23225
+        BUILD_PARAM="${BUILD_PARAM} --disable_mkl_dnn"
         ;;
 esac