experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Nightly breakage - Explicitly skip gpu builds.

Browse Source 
PiperOrigin-RevId: 350016772
Change-Id: I047ff5c8c0d969cb4c7a4fd5f8f304773b468c4f
This commit is contained in:
Hye Soo Yang 2021-01-04 07:40:06 -08:00 committed by TensorFlower Gardener
parent d18947de13
commit 556fa126db
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/kernel_tests/extract_volume_patches_grad_test.py
View File

@ -58,6 +58,9 @@ class ExtractVolumePatchesGradTest(test.TestCase, parameterized.TestCase):
}, },
]) ])
def testGradient(self, in_shape, ksizes, strides): 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. # Set graph seed for determinism.
random_seed = 42 random_seed = 42
random_seed_lib.set_random_seed(random_seed) random_seed_lib.set_random_seed(random_seed)