Skip to content

Commit 4393e95

Browse files
Merge pull request tensorflow#28620 from adamjrichter:master
PiperOrigin-RevId: 248230390
2 parents 158cf19 + 4bc2202 commit 4393e95

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tensorflow/core/kernels/eigen_backward_spatial_convolutions.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,8 +366,8 @@ SpatialConvolutionBackwardKernel(
366366
YOU_MADE_A_PROGRAMMING_MISTAKE);
367367

368368
// stride and in_stride cannot both be larger than 1
369-
eigen_assert(!(row_stride > 1 && row_in_stride > 1) &&
370-
!(col_stride > 1 && col_in_stride > 1));
369+
eigen_assert(!(row_stride > 1 && row_in_stride > 1));
370+
eigen_assert(!(col_stride > 1 && col_in_stride > 1));
371371

372372
static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor);
373373

tensorflow/core/util/gpu_device_functions.h

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -459,7 +459,8 @@ __host__ __device__ inline std::complex<double> CudaLdg(
459459
template <typename T>
460460
__global__ void SetZero(const int count, T* ptr) {
461461
// Check that the grid is one dimensional and index doesn't overflow.
462-
assert(blockDim.y == 1 && blockDim.z == 1);
462+
assert(blockDim.y == 1);
463+
assert(blockDim.z == 1);
463464
assert(blockDim.x * gridDim.x / blockDim.x == gridDim.x);
464465
for (int i : CudaGridRangeX(count)) {
465466
ptr[i] = T(0);
@@ -470,7 +471,8 @@ __global__ void SetZero(const int count, T* ptr) {
470471
template <typename T>
471472
__global__ void SetToValue(const int count, T* ptr, T value) {
472473
// Check that the grid is one dimensional and index doesn't overflow.
473-
assert(blockDim.y == 1 && blockDim.z == 1);
474+
assert(blockDim.y == 1);
475+
assert(blockDim.z == 1);
474476
assert(blockDim.x * gridDim.x / blockDim.x == gridDim.x);
475477
for (int i : CudaGridRangeX(count)) {
476478
ptr[i] = value;

0 commit comments

Comments
 (0)