Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 3c2d342

Browse files
committed
enable integer reduction on windows
1 parent d306d28 commit 3c2d342

File tree

1 file changed

+0
-13
lines changed

1 file changed

+0
-13
lines changed

src/operator/tensor/broadcast_reduce-inl.h

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -241,8 +241,6 @@ void Reduce(Stream<cpu>* s, const TBlob& small, const OpReqType req,
241241
N, M, req == kAddTo, big.dptr<DType>(), small.dptr<DType>(),
242242
big.shape_.get<ndim>(), small.shape_.get<ndim>(), rshape, rstride);
243243
} else {
244-
// TODO(haojin2): Use real-only type swtich for windows temporarily due to CI issues.
245-
#ifndef _WIN32
246244
MXNET_ACC_TYPE_SWITCH(mshadow::DataType<DType>::kFlag, DataType, AType, {
247245
typedef typename std::conditional<safe_acc, AType, DataType>::type AccType;
248246
MSHADOW_TYPE_SWITCH_WITH_BOOL(small.type_flag_, OType, {
@@ -252,17 +250,6 @@ void Reduce(Stream<cpu>* s, const TBlob& small, const OpReqType req,
252250
big.shape_.get<ndim>(), small.shape_.get<ndim>(), rshape, rstride);
253251
});
254252
});
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
266253
}
267254
}
268255

0 commit comments

Comments
 (0)