Nightly breakage - Explicitly skip gpu builds.

PiperOrigin-RevId: 350016772 Change-Id: I047ff5c8c0d969cb4c7a4fd5f8f304773b468c4f
2021-01-04 07:40:06 -08:00 · 2021-01-04 07:40:06 -08:00 · 556fa126db
commit 556fa126db
parent d18947de13
1 changed files with 3 additions and 0 deletions
--- a/tensorflow/python/kernel_tests/extract_volume_patches_grad_test.py
+++ b/tensorflow/python/kernel_tests/extract_volume_patches_grad_test.py
@ -58,6 +58,9 @@ class ExtractVolumePatchesGradTest(test.TestCase, parameterized.TestCase):
      },
  ])
  def testGradient(self, in_shape, ksizes, strides):
+    if test_util.is_gpu_available():
+      self.skipTest('b/171837334: skip gpu test.')
+
    # Set graph seed for determinism.
    random_seed = 42
    random_seed_lib.set_random_seed(random_seed)