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

Move MKLDNN debug message under MXNET_MKLDNN_DEBUG #13662

Merged
merged 2 commits into from
Dec 20, 2018
Merged

Move MKLDNN debug message under MXNET_MKLDNN_DEBUG #13662

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
881b945
Move the debug output message into MXNET_MKLDNN_DEBUG
pengzhao-intel Dec 17, 2018
ba57334
Merge remote-tracking branch 'official/master' into remove-malloc-info
pengzhao-intel Dec 18, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion src/operator/nn/mkldnn/mkldnn_base.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -70,9 +70,10 @@ mkldnn::memory *TmpMemMgr::Alloc(const mkldnn::memory::primitive_desc &pd) {
} else {
// If curr_mem has been initialized and we still reach here. It means
// the current allocated memory isn't enough.
if (this->curr_mem)
if (this->curr_mem && dmlc::GetEnv("MXNET_MKLDNN_DEBUG", false)) {
LOG(WARNING) << "Allocate " << pd.get_size()
<< " bytes with malloc directly";
}
mkldnn_mem_ptr ret(new mkldnn::memory(pd));
MKLDNNStream::Get()->RegisterMem(ret);
return ret.get();
Expand Down