Fix fft_ops_test.py for CPU

2017-06-08 11:58:21 -07:00 · 2017-06-08 11:58:21 -07:00 · 50d80ddf92
commit 50d80ddf92
parent 88d648f3be
1 changed files with 27 additions and 26 deletions
--- a/tensorflow/python/kernel_tests/fft_ops_test.py
+++ b/tensorflow/python/kernel_tests/fft_ops_test.py
@ -298,6 +298,7 @@ class RFFTOpsTest(BaseFFTOpsTest):
                                use_placeholder=True)

  def testFftLength(self):
+    if test.is_gpu_available(cuda_only=True):
      for rank in VALID_FFT_RANKS:
        for dims in xrange(rank, rank + 3):
          for size in (5, 6):