saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Logging in framework.cc should use the session logger (#3059)

Browse source
This commit is contained in:
Yufeng Li 2020-02-21 17:11:14 -08:00 committed by GitHub
parent fb871978b5
commit 64feee1b52
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
onnxruntime/core/framework/execution_frame.cc
View file

@ -290,13 +290,13 @@ Status ExecutionFrame::AllocateMLValueTensorSelfOwnBufferHelper(OrtValue& ort_va
// fed in, so use VERBOSE as the log level as it's expected.
// TODO: Should we re-use the block if the size is large enough? Would probably need to allow it
// to be freed if the size difference was too large so our memory usage doesn't stick at a high water mark
LOGS_DEFAULT(VERBOSE) << "For ort_value with index: " << ort_value_index
<< ", block in memory pattern size is: " << block->size_
<< " but the actually size is: " << size
<< ", fall back to default allocation behavior";
LOGS(session_state_.Logger(), VERBOSE) << "For ort_value with index: " << ort_value_index
<< ", block in memory pattern size is: " << block->size_
<< " but the actually size is: " << size
<< ", fall back to default allocation behavior";
} else if (it == buffers_.end()) {
LOGS_DEFAULT(WARNING) << "For ort_value with index: " << ort_value_index
<< ", block not found in target location. fall back to default allocation behavior";
LOGS(session_state_.Logger(), WARNING) << "For ort_value with index: " << ort_value_index
<< ", block not found in target location. fall back to default allocation behavior";
}
}
}
@ -340,7 +340,7 @@ Status ExecutionFrame::AllocateMLValueTensorPreAllocateBuffer(OrtValue& ort_valu
// be generous and use the buffer if it's large enough. log a warning though as it indicates a bad model
if (buffer_num_elements >= required_num_elements) {
LOGS_DEFAULT(WARNING) << message;
LOGS(session_state_.Logger(), WARNING) << message;
} else {
return ORT_MAKE_STATUS(ONNXRUNTIME, FAIL, message);
}
@ -377,7 +377,7 @@ static Status AllocateTraditionalMLValue(OrtValue& ort_value, const NonTensorTyp
return Status::OK();
}
static Status AllocateTensorSequence (OrtValue& ort_value) {
static Status AllocateTensorSequence(OrtValue& ort_value) {
auto ml_tensor_sequence = DataTypeImpl::GetType<TensorSeq>();
auto p_tensor_sequence = onnxruntime::make_unique<TensorSeq>();
ort_value.Init(p_tensor_sequence.release(), ml_tensor_sequence, ml_tensor_sequence->GetDeleteFunc());