From 1cd4b5c7edd5c529df00fc3149112426ddd9c28f Mon Sep 17 00:00:00 2001
From: Nitin Srinivasan <srnitin@google.com>
Date: Fri, 8 Dec 2023 17:39:50 -0800
Subject: [PATCH] Disable `matrix_band_part_test` on Arm64 machines

This is intended to be temporary to unblock the Linux Arm64 GitHub presubmit.

PiperOrigin-RevId: 589287236
---
 tensorflow/compiler/tests/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD
index 0fb97c4dc9f96f..ae803f5d16dd04 100644
--- a/tensorflow/compiler/tests/BUILD
+++ b/tensorflow/compiler/tests/BUILD
@@ -1017,6 +1017,7 @@ tf_xla_py_strict_test(
     enable_mlir_bridge = True,
     python_version = "PY3",
     tags = [
+        "no_aarch64",  # TODO(b/315533266)
         "no_pip",  # TODO(b/149738646): fix pip install so these tests run on kokoro pip
         "optonly",
     ],