format

apache · Dec 9, 2019 · 176fc22 · 176fc22
1 parent 8aff047
commit 176fc22
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/src/operator/numpy/np_bincount_op-inl.h b/src/operator/numpy/np_bincount_op-inl.h
@@ -53,8 +53,8 @@ struct NumpyBincountParam : public dmlc::Parameter<NumpyBincountParam> {
 };
 
 inline bool NumpyBincountType(const nnvm::NodeAttrs& attrs,
-                     std::vector<int> *in_attrs,
-                     std::vector<int> *out_attrs) {
+                              std::vector<int> *in_attrs,
+                              std::vector<int> *out_attrs) {
   const NumpyBincountParam& param = nnvm::get<NumpyBincountParam>(attrs.parsed);
   if (!param.has_weights) {
     return ElemwiseType<1, 1>(attrs, in_attrs, out_attrs) && in_attrs->at(0) != -1;

diff --git a/src/operator/numpy/np_bincount_op.cu b/src/operator/numpy/np_bincount_op.cu
@@ -144,13 +144,13 @@ void NumpyBincountForwardImpl<gpu>(const OpContext &ctx,
   });
 
   MSHADOW_TYPE_SWITCH(data.dtype(), DType, {
-      MSHADOW_TYPE_SWITCH(out.dtype(), OType, {
-        size_t out_size = out.shape().Size();
-        Kernel<set_zero, gpu>::Launch(s, out_size, out.data().dptr<OType>());
-        Kernel<BincountFusedKernel, gpu>::Launch(
-          s, data_n, data.data().dptr<DType>(), out.data().dptr<OType>());
-      });
+    MSHADOW_TYPE_SWITCH(out.dtype(), OType, {
+      size_t out_size = out.shape().Size();
+      Kernel<set_zero, gpu>::Launch(s, out_size, out.data().dptr<OType>());
+      Kernel<BincountFusedKernel, gpu>::Launch(
+        s, data_n, data.data().dptr<DType>(), out.data().dptr<OType>());
     });
+  });
 }
 
 NNVM_REGISTER_OP(_npi_bincount)