change back to MSHADOW_TYPE_SWITCH

apache · Apr 11, 2019 · bb297b8 · bb297b8
1 parent cc6a684
commit bb297b8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/operator/tensor/broadcast_reduce_op.h b/src/operator/tensor/broadcast_reduce_op.h
@@ -565,13 +565,12 @@ void ReduceAxesComputeImpl(const OpContext& ctx,
   mxnet::TShape src_shape, dst_shape;
   BroadcastReduceShapeCompact(inputs[0].shape_, small, &src_shape, &dst_shape);
   Stream<xpu> *s = ctx.get_stream<xpu>();
-  MXNET_ACC_TYPE_SWITCH(inputs[0].type_flag_, DType, AType, {
+  MSHADOW_TYPE_SWITCH(inputs[0].type_flag_, DType, {
     MSHADOW_TYPE_SWITCH(outputs[0].type_flag_, OType, {
       const TBlob in_data = inputs[0].reshape(src_shape);
       const TBlob out_data = outputs[0].reshape(dst_shape);
       BROADCAST_NDIM_SWITCH(dst_shape.ndim(), NDim, {
-        typedef typename std::conditional<safe_acc, AType, DType>::type AccType;
-        size_t workspace_size = broadcast::ReduceWorkspaceSize<NDim, AccType>(
+        size_t workspace_size = broadcast::ReduceWorkspaceSize<NDim, DType>(
             s, out_data.shape_, req[0], in_data.shape_);
         Tensor<xpu, 1, char> workspace =
             ctx.requested[0].get_space_typed<xpu, 1, char>(Shape1(workspace_size), s);