Fix incorrect merge of #34136. (#36760)

gchanan · facebook-github-bot · commit ebdc4f02ad5d · 2020-04-17T07:25:13.000-07:00
Summary: Pull Request resolved: #36760 If you look at #34136, you will notice a commit (80c15c0) that didn't get merged. This is to address that, to avoid crashing on remainder when the rhs is 0. Test Plan: Imported from OSS Differential Revision: D21078776 Pulled By: gchanan fbshipit-source-id: 0ac138cbafac28cf8d696a2a413d3c542138cff9
diff --git a/aten/src/ATen/native/cuda/BinaryArithmeticKernel.cu b/aten/src/ATen/native/cuda/BinaryArithmeticKernel.cu
@@ -70,7 +70,6 @@ void remainder_kernel_cuda(TensorIterator& iter) {
     AT_DISPATCH_INTEGRAL_TYPES(iter.dtype(), "remainder_cuda", [&]() {
       using thrust_t = typename ztype_cuda<scalar_t>::thrust_t;
       gpu_kernel_with_scalars(iter, []GPU_LAMBDA(thrust_t a, thrust_t b) -> thrust_t {
-        CUDA_KERNEL_ASSERT(b != 0);
         thrust_t r = a % b;
         if ((r != 0) && ((r < 0) != (b < 0))) {
           r += b;
