frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-05 08:24:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

net benchmark code

Browse Source
This commit is contained in:
Yangqing Jia
2015-12-15 20:42:22 -08:00
parent cf8ffe215f
commit 6b27cabf17
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
caffe2/core/net.cc
View File

@ -87,17 +87,31 @@ void SimpleNet::TEST_Benchmark(const int warmup_runs, const int main_runs,
CAFFE_LOG_INFO << "Main run finished. Milliseconds per iter: " CAFFE_LOG_INFO << "Main run finished. Milliseconds per iter: "
<< timer.MilliSeconds() / main_runs; << timer.MilliSeconds() / main_runs;
int idx = 0; int idx = 0;
float total = 0.;
CaffeMap<string, float> time_per_op_type;
if (run_individual) { if (run_individual) {
for (auto& op : operators_) { for (auto& op : operators_) {
const string& op_type = op->def().type();
CAFFE_LOG_INFO << "Running operator #" << idx << " (" CAFFE_LOG_INFO << "Running operator #" << idx << " ("
<< op->def().name() << ", " << op->def().type() << op->def().name() << ", " << op_type << ")";
<< ")";
timer.Start(); timer.Start();
for (int i = 0; i < main_runs; ++i) { for (int i = 0; i < main_runs; ++i) {
CAFFE_CHECK(op->Run()); CAFFE_CHECK(op->Run());
} }
float spent = timer.MilliSeconds() / main_runs;
total += spent;
CAFFE_LOG_INFO << " Finished. Milliseconds per iter: " CAFFE_LOG_INFO << " Finished. Milliseconds per iter: "
<< timer.MilliSeconds() / main_runs; << spent;
//if (time_per_op_type.count(op_type) == 0) {
// time_per_op_type[op_type] = 0.;
//}
time_per_op_type[op_type] += spent;
++idx;
}
CAFFE_LOG_INFO << "Per-op run total: " << total;
CAFFE_LOG_INFO << "Time per operator type:";
for (const auto& item : time_per_op_type) {
CAFFE_LOG_INFO << "\t" << item.second << "\t" << item.first;
} }
} }
} }