Skip to content

[edge profiler] Add support for performance events counting #87876

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 11 commits into from
Closed

[edge profiler] Add support for performance events counting #87876

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
31d5ff2
[edge profiler] Add support for performance events counting
digantdesai Oct 27, 2022
218af12
Update on "[edge profiler] Add support for performance events counting"
digantdesai Oct 27, 2022
51c6353
Update on "[edge profiler] Add support for performance events counting"
digantdesai Oct 27, 2022
9563646
Update on "[edge profiler] Add support for performance events counting"
digantdesai Oct 28, 2022
4b7347f
Update on "[edge profiler] Add support for performance events counting"
digantdesai Oct 28, 2022
6738404
Update on "[edge profiler] Add support for performance events counting"
digantdesai Oct 31, 2022
0e07a12
Update on "[edge profiler] Add support for performance events counting"
digantdesai Nov 1, 2022
bcbbcd7
Update on "[edge profiler] Add support for performance events counting"
digantdesai Nov 1, 2022
03864e7
Update on "[edge profiler] Add support for performance events counting"
digantdesai Nov 1, 2022
2015a8f
Update on "[edge profiler] Add support for performance events counting"
digantdesai Nov 1, 2022
a55fdaa
Update on "[edge profiler] Add support for performance events counting"
digantdesai Nov 2, 2022
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
12 changes: 10 additions & 2 deletions torch/csrc/jit/mobile/profiler_edge.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,23 @@ KinetoEdgeCPUProfiler::KinetoEdgeCPUProfiler(
const bool profile_memory,
const bool with_stack,
const bool with_flops,
const bool with_modules)
const bool with_modules,
std::vector<std::string> events)
: m_(m), trace_file_name_(fname) {
torch::profiler::impl::ExperimentalConfig experimental_config;
// Enable hardware counters
if (events.size()) {
experimental_config.performance_events = std::move(events);
}

torch::profiler::impl::ProfilerConfig config(
torch::profiler::impl::ProfilerState::KINETO,
report_input_shapes,
profile_memory,
with_stack,
with_flops,
with_modules);
with_modules,
experimental_config);
torch::autograd::profiler::prepareProfiler(
config, {torch::autograd::profiler::ActivityType::CPU});
if (with_modules || with_stack) {
Expand Down
3 changes: 2 additions & 1 deletion torch/csrc/jit/mobile/profiler_edge.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,8 @@ class TORCH_API KinetoEdgeCPUProfiler {
const bool profile_memory = false,
const bool with_stack = false,
const bool with_flops = false,
const bool with_modules = false);
const bool with_modules = false,
std::vector<std::string> events = {});

const std::unique_ptr<torch::autograd::profiler::ProfilerResult>&
disableProfiler();
Expand Down