Simplify complex case for div_cpu (#39996)

ShawnZhong · facebook-github-bot · commit 79736ff9c206 · 2020-06-23T08:48:10.000-07:00
Summary: Simplify complex case for `div_cpu` cc: zasdfgbnm Pull Request resolved: #39996 Differential Revision: D22169715 Pulled By: anjali411 fbshipit-source-id: e1822ee5c575cb8786b395c1bc7550890b38a60d
diff --git a/aten/src/ATen/native/cpu/BinaryOpsKernel.cpp b/aten/src/ATen/native/cpu/BinaryOpsKernel.cpp
@@ -76,26 +76,16 @@ void div_kernel(TensorIterator& iter) {
         return a / b;
       });
     });
-  } else if (isComplexType(iter.dtype())) {
-      AT_DISPATCH_COMPLEX_TYPES(iter.dtype(), "div_cpu", [&]() {
-        cpu_kernel_vec(iter,
-          [](scalar_t a, scalar_t b) __ubsan_ignore_float_divide_by_zero__ -> scalar_t {
-             return a / b;
-          },
-          [](Vec256<scalar_t> a, Vec256<scalar_t> b) {
-            return a / b;
-          });
-      });
-    } else {
-      AT_DISPATCH_FLOATING_TYPES_AND2(kBFloat16, kHalf, iter.dtype(), "div_cpu", [&]() {
-        cpu_kernel_vec(iter,
-          [=](scalar_t a, scalar_t b) __ubsan_ignore_float_divide_by_zero__ -> scalar_t {
-            return a / b;
-          },
-          [=](Vec256<scalar_t> a, Vec256<scalar_t> b) {
-            return a / b;
-          });
-      });
+  } else {
+    AT_DISPATCH_FLOATING_AND_COMPLEX_TYPES_AND2(kBFloat16, kHalf, iter.dtype(), "div_cpu", [&]() {
+      cpu_kernel_vec(iter,
+        [](scalar_t a, scalar_t b) __ubsan_ignore_float_divide_by_zero__ -> scalar_t {
+          return a / b;
+        },
+        [](Vec256<scalar_t> a, Vec256<scalar_t> b) {
+          return a / b;
+        });
+    });
   }
 }
 
