OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[SR][C2][easy] Benchmarks report # of ops (#67436)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67436

This information is useful for comparing static runtime to c2

Reviewed By: d1jang

Differential Revision: D31991571

fbshipit-source-id: eb83bc4564b05d56fb9a550863eea3f6312f3f6c
This commit is contained in:
Mike Iovine 2021-10-28 13:01:23 -07:00 committed by Facebook GitHub Bot
parent b8f07689f2
commit 8363da3f92
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
caffe2/core/net_simple.cc
View File

@ -322,6 +322,7 @@ void IndividualMetrics::PrintOperatorProfilingResults() {
<< output_dealloc_time * normalizer[i] << " " << unit[i]
<< std::endl;
}
std::cout << "Number of operators: " << operators_.size() << std::endl;
}
}
}

1
torch/csrc/jit/runtime/static/impl.cpp
View File

@ -1060,6 +1060,7 @@ void StaticRuntime::benchmark(
<< " ms" << std::endl;
std::cout << "First iter time: " << results.first_iter_time << " ms"
<< std::endl;
std::cout << "Number of operators: " << nodes_.size() << std::endl;
if (planner_) {
std::cout << "Total number of managed tensors: "