Skip to content

Commit 8fb9fe2

Browse files
ctcyangJose Luis Contreras
authored and
Jose Luis Contreras
committed
[MXNET-1111] Remove CPUPinned in ImageRecordIter (apache#12666)
* squash commit * get rid of argument * undo a lot of unnecessary changes * undo more changes * fix typo * fix lint * address comments and fix rebase mistake * fix typo made during rebase * revert cpu_pinned * revert changes, because works without needing to copy params to GPU. thanks @yuxihu for testing and @apeforest for raising this issue! * revert changes to comm and nccl
1 parent 6350742 commit 8fb9fe2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/io/iter_image_recordio_2.cc

+2-2
Original file line numberDiff line numberDiff line change
@@ -285,9 +285,9 @@ inline bool ImageRecordIOParser2<DType>::ParseNext(DataBatch *out) {
285285
shape_vec.push_back(param_.label_width);
286286
TShape label_shape(shape_vec.begin(), shape_vec.end());
287287

288-
out->data.at(0) = NDArray(data_shape, Context::CPUPinned(0), false,
288+
out->data.at(0) = NDArray(data_shape, Context::CPU(0), false,
289289
mshadow::DataType<DType>::kFlag);
290-
out->data.at(1) = NDArray(label_shape, Context::CPUPinned(0), false,
290+
out->data.at(1) = NDArray(label_shape, Context::CPU(0), false,
291291
mshadow::DataType<real_t>::kFlag);
292292
unit_size_[0] = param_.data_shape.Size();
293293
unit_size_[1] = param_.label_width;

0 commit comments

Comments
 (0)