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

Commit

Permalink
create copy of arrays for fallback
Browse files Browse the repository at this point in the history
  • Loading branch information
azai91 committed Aug 30, 2018
1 parent de9cffa commit f3e55e9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/operator/nn/mkldnn/mkldnn_base-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -361,9 +361,9 @@ static inline void CreateDefaultInputs(const std::vector<NDArray> &arrs,
std::vector<NDArray> *out_arrs) {
for (size_t i = 0; i < arrs.size(); ++i) {
if (arrs[i].IsMKLDNNData())
out_arrs->emplace_back(arrs[i].Reorder2Default());
out_arrs->push_back(arrs[i].Reorder2Default());
else
out_arrs->emplace_back(arrs[i]);
out_arrs->push_back(arrs[i]);
}
}

Expand Down

0 comments on commit f3e55e9

Please sign in to comment.