Adjustments to fft_ops_test.py.

PiperOrigin-RevId: 158612165
TeeeJay · Jun 10, 2017 · 51ec66e · 51ec66e
1 parent 88d4076
commit 51ec66e
Show file tree

Hide file tree

Showing 3 changed files with 225 additions and 183 deletions.
diff --git a/tensorflow/core/kernels/fft_ops.cc b/tensorflow/core/kernels/fft_ops.cc
@@ -206,10 +206,12 @@ class FFTCPU : public FFTBase {
  reversedAxis[i] = i == FFTRank;
  }
 
- full_fft.slice(negTargetIndices, negSizes) =
- full_fft.slice(negStartIndices, negSizes)
- .reverse(reversedAxis)
- .conjugate();
+ if (negSizes[FFTRank] != 0) {
+ full_fft.slice(negTargetIndices, negSizes) =
+ full_fft.slice(negStartIndices, negSizes)
+ .reverse(reversedAxis)
+ .conjugate();
+ }
 
  auto innerAxis = Eigen::array<int, 1>{FFTRank};
  output.device(device) =

diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
@@ -2184,11 +2184,7 @@ cuda_py_test(
  "//tensorflow/python:math_ops",
  "//tensorflow/python:spectral_ops",
  ],
- tags = [
- "noasan",
- "nomsan",
- "notsan", # http:https://b/62067589
- ],
+ shard_count = 3,
 )
 
 cuda_py_test(