benchmark
9b1ac22b - Add --profile-options to support fine toggling of profile options (#1306)

Commit
3 years ago
Add --profile-options to support fine toggling of profile options (#1306) Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/1306 Support a new flag, --profile-options, to enable fine grained toggling of profiler options, such as record_shapes, profile_memory, with_stack, with_flops, and with_modules. This allows the user to pick which options to enable rather than using --profile-detailed which enables all options. Test Plan: CI Tests Reviewed By: xuzhao9 Differential Revision: D41314079 Pulled By: aaronenyeshi fbshipit-source-id: d8eea28ad79e6ebd830941653ec6491742cbcbb0
Author
Parents
Loading