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

Misc. cleanup. (#5659)

Browse source 
Co-authored-by: Ubuntu <OrtTrainingDev3@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
This commit is contained in:
M. Zeeshan Siddiqui 2020-11-02 07:05:28 -08:00 committed by GitHub
parent 9af0d48524
commit f2168cef29
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -292,7 +292,7 @@ ExecutionFrame::ExecutionFrame(const std::vector<int>& feed_mlvalue_idxs, const
}
// log size of activation. Keep it commented out for now to avoid log flooding.
printf("\n **** Allocated memory for activations, size: %zu ***\n", mem_patterns_->patterns[i].PeakSize());
VLOGS(session_state_.Logger(), 1) << "**** Allocated memory for activations, size: " <<mem_patterns_->patterns[i].PeakSize();
}
}
}

9
onnxruntime/core/framework/mem_pattern_planner.h
View file

@ -42,24 +42,21 @@ class MemPatternPlanner {
size_t index_1 = 0;
size_t index_2 = 0;
bool overlap = false;
while ((index_1 < program_counter_start_1.size()) && (index_2 < program_counter_start_2.size())) {
if (program_counter_start_1[index_1] <= program_counter_start_2[index_2]) {
if (program_counter_end_1[index_1] >= program_counter_start_2[index_2]) {
overlap = true;
break;
return true;
}
index_1 += 1;
} else {
if (program_counter_end_2[index_2] >= program_counter_start_1[index_1]) {
overlap = true;
break;
return true;
}
index_2 += 1;
}
}
return overlap;
return false;
}
void TraceAllocation(int ml_value_idx, const std::vector<size_t>& program_counter_start, const std::vector<size_t>& program_counter_end, size_t size) {

2
orttraining/tools/ci_test/run_convergence_test.py
View file

@ -31,8 +31,6 @@ def main():
# run BERT training
subprocess.run([
"/bert_ort/openmpi/bin/mpirun",
"-n", "2",
os.path.join(args.binary_dir, "onnxruntime_training_bert"),
"--model_name", os.path.join(
args.model_root, "nv/bert-base/bert-base-uncased_L_12_H_768_A_12_V_30528_S_512_Dp_0.1_optimized_layer_norm_opset12"),