From 1c5d7d423cd47077bbdc624e950e616a56b64564 Mon Sep 17 00:00:00 2001 From: Milica Makevic Date: Thu, 7 Nov 2024 09:58:14 +0000 Subject: [PATCH] Disable svd tests --- tensorflow/compiler/tests/BUILD | 3 ++- third_party/xla/xla/hlo/builder/lib/BUILD | 5 ++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index feff4af477cbbe..c5965d94248c7f 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -571,7 +571,8 @@ tf_xla_py_strict_test( python_version = "PY3", tags = [ "no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip - "optonly" + "optonly", + "cuda-only" #TODO(rocm): weekly sync 24-11-05 ], deps = [ ":xla_test", diff --git a/third_party/xla/xla/hlo/builder/lib/BUILD b/third_party/xla/xla/hlo/builder/lib/BUILD index cfa34fb7cf8bcf..b2a4163f350e99 100644 --- a/third_party/xla/xla/hlo/builder/lib/BUILD +++ b/third_party/xla/xla/hlo/builder/lib/BUILD @@ -646,7 +646,10 @@ xla_test( srcs = ["svd_test.cc"], real_hardware_only = True, shard_count = 10, - tags = ["optonly"], + tags = [ + "optonly", + "cuda-only" #TODO(rocm): weekly sync 24-11-05 + ], deps = [ ":arithmetic", ":constants",