diff --git a/plugin/xprof/profile_plugin.py b/plugin/xprof/profile_plugin.py index 2815c599..23c92104 100644 --- a/plugin/xprof/profile_plugin.py +++ b/plugin/xprof/profile_plugin.py @@ -1530,6 +1530,7 @@ def capture_route_impl(self, request: wrappers.Request) -> wrappers.Response: request.args.get('python_tracer_level', '0') ), 'delay_ms': int(request.args.get('delay', '0')), + 'enable_hlo_proto': True, } if is_tpu_name: