@@ -241,8 +241,6 @@ void Reduce(Stream<cpu>* s, const TBlob& small, const OpReqType req,
241
241
N, M, req == kAddTo , big.dptr <DType>(), small.dptr <DType>(),
242
242
big.shape_ .get <ndim>(), small.shape_ .get <ndim>(), rshape, rstride);
243
243
} else {
244
- // TODO(haojin2): Use real-only type swtich for windows temporarily due to CI issues.
245
- #ifndef _WIN32
246
244
MXNET_ACC_TYPE_SWITCH (mshadow::DataType<DType>::kFlag , DataType, AType, {
247
245
typedef typename std::conditional<safe_acc, AType, DataType>::type AccType;
248
246
MSHADOW_TYPE_SWITCH_WITH_BOOL (small.type_flag_ , OType, {
@@ -252,17 +250,6 @@ void Reduce(Stream<cpu>* s, const TBlob& small, const OpReqType req,
252
250
big.shape_ .get <ndim>(), small.shape_ .get <ndim>(), rshape, rstride);
253
251
});
254
252
});
255
- #else
256
- MXNET_REAL_ACC_TYPE_SWITCH (mshadow::DataType<DType>::kFlag , DataType, AType, {
257
- typedef typename std::conditional<safe_acc, AType, DataType>::type AccType;
258
- MSHADOW_TYPE_SWITCH_WITH_BOOL (small.type_flag_ , OType, {
259
- typedef typename std::conditional<safe_acc, OType, DataType>::type OutType;
260
- seq_reduce_compute<Reducer, ndim, AccType, DataType, OutType, OP>(
261
- N, M, req == kAddTo , big.dptr <DataType>(), small.dptr <OutType>(),
262
- big.shape_ .get <ndim>(), small.shape_ .get <ndim>(), rshape, rstride);
263
- });
264
- });
265
- #endif
266
253
}
267
254
}
268
255
0 commit comments