From 3c2d342c7d8a72e68e566fc90555c1a7afd67cb1 Mon Sep 17 00:00:00 2001 From: Hao Jin Date: Wed, 13 Nov 2019 01:38:48 +0000 Subject: [PATCH] enable integer reduction on windows --- src/operator/tensor/broadcast_reduce-inl.h | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/src/operator/tensor/broadcast_reduce-inl.h b/src/operator/tensor/broadcast_reduce-inl.h index 800dbffbcd38..e203da2a127e 100644 --- a/src/operator/tensor/broadcast_reduce-inl.h +++ b/src/operator/tensor/broadcast_reduce-inl.h @@ -241,8 +241,6 @@ void Reduce(Stream* s, const TBlob& small, const OpReqType req, N, M, req == kAddTo, big.dptr(), small.dptr(), big.shape_.get(), small.shape_.get(), rshape, rstride); } else { - // TODO(haojin2): Use real-only type swtich for windows temporarily due to CI issues. -#ifndef _WIN32 MXNET_ACC_TYPE_SWITCH(mshadow::DataType::kFlag, DataType, AType, { typedef typename std::conditional::type AccType; MSHADOW_TYPE_SWITCH_WITH_BOOL(small.type_flag_, OType, { @@ -252,17 +250,6 @@ void Reduce(Stream* s, const TBlob& small, const OpReqType req, big.shape_.get(), small.shape_.get(), rshape, rstride); }); }); -#else - MXNET_REAL_ACC_TYPE_SWITCH(mshadow::DataType::kFlag, DataType, AType, { - typedef typename std::conditional::type AccType; - MSHADOW_TYPE_SWITCH_WITH_BOOL(small.type_flag_, OType, { - typedef typename std::conditional::type OutType; - seq_reduce_compute( - N, M, req == kAddTo, big.dptr(), small.dptr(), - big.shape_.get(), small.shape_.get(), rshape, rstride); - }); - }); -#endif } }