Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
TaoLv committed Sep 9, 2019
1 parent e73764f commit 8cc4201
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
3 changes: 1 addition & 2 deletions src/ndarray/ndarray.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1078,7 +1078,7 @@ inline void CopyFromToRspImpl(const NDArray& from, const NDArray& to, RunContext
// Make a copy of a dense NDArray
template<typename from_xpu, typename to_xpu>
inline void CopyFromToDnsImpl(const NDArray& from, const NDArray& to, RunContext ctx) {
#if MXNET_USE_MKLDNN ==100
#if MXNET_USE_MKLDNN == 100
// If neither is MKLDNN, we can copy data normally.
if (!from.IsMKLDNNData() && !to.IsMKLDNNData()) {
#endif
Expand Down Expand Up @@ -1616,7 +1616,6 @@ void NDArray::Save(dmlc::Stream *strm) const {
#if MXNET_USE_MKLDNN == 1
if (nd_cpu.IsMKLDNNData()) {
LOG(FATAL) << "TODO: MKL-DNN 1.0";

nd_cpu = nd_cpu.Reorder2Default();
}
#endif
Expand Down
2 changes: 1 addition & 1 deletion src/operator/nn/mkldnn/mkldnn_base-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -352,7 +352,7 @@ class MKLDNNStream {
public:
static MKLDNNStream *Get();

MKLDNNStream(): s(CpuEngine::Get()->get_engine()) {};
MKLDNNStream(): s(CpuEngine::Get()->get_engine()) {}

void RegisterPrimArgs(const mkldnn::primitive &prim,
const mkldnn_args_map_t &args) {
Expand Down

0 comments on commit 8cc4201

Please sign in to comment.