Include resource managers in pathwaysutils profile request by default. - #325
Conversation
|
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). View this failed invocation of the CLA check for more information. For the most up to date status, view the checks section at the bottom of the pull request. |
There was a problem hiding this comment.
Code Review
This pull request introduces a new boolean parameter include_resource_managers (defaulting to True) across the profiling utility functions, including start_trace, collect_profile, and the FastAPI ProfilingConfig. This allows users to control whether resource manager profiling is included in the JAX profile request. The corresponding unit tests have been updated and expanded to verify this new configuration option. There are no review comments, so I have no feedback to provide.
3da13a0 to
f9877dc
Compare
f9877dc to
1ed19fd
Compare
1ed19fd to
e1fd75c
Compare
e1fd75c to
ce2d1b9
Compare
PiperOrigin-RevId: 982600443
ce2d1b9 to
67c531d
Compare
Include resource managers in pathwaysutils profile request by default.