tbp-nightly 2.20.0a20250203__cp311-none-manylinux2014_x86_64.whl → 2.20.0a20250205__cp311-none-manylinux2014_x86_64.whl

{tbp_nightly-2.20.0a20250203.dist-info → tbp_nightly-2.20.0a20250205.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tbp-nightly
-Version: 2.20.0a20250203
+Version: 2.20.0a20250205
 Summary: Profile Tensorboard Plugin
 Home-page: https://github.com/tensorflow/profiler
 Author: Google Inc.

{tbp_nightly-2.20.0a20250203.dist-info → tbp_nightly-2.20.0a20250205.dist-info}/RECORD

@@ -1,15 +1,15 @@
 tensorboard_plugin_profile/__init__.py,sha256=9odmeTOiGoJv2r3-l3N3SnAllQMKVIw6rgX4BxOnB-4,923
 tensorboard_plugin_profile/profile_plugin.py,sha256=R4QwoYnqLxyU5NADaN0PZb2pYF3aj3mC4a7CGeVAvJo,34978
 tensorboard_plugin_profile/profile_plugin_loader.py,sha256=7EbITLel-uFo1ZB28Kv4CFxBeSRzzEeZL6FEf6SbUpQ,2873
-tensorboard_plugin_profile/version.py,sha256=I9BWoSph62dXu1J_anPm5sd8S2aSDnmLpNuQmkRaupw,780
-tensorboard_plugin_profile/convert/_pywrap_profiler_plugin.so,sha256=AmXk_Tz9r7r5J07TvbzDhQ_o2kaB4yn82YMJkaGpgoE,98724328
+tensorboard_plugin_profile/version.py,sha256=DXSJ5HNlXncwjZCD7ENXxWHxYHdx8U-lFIqeMA8O2xA,780
+tensorboard_plugin_profile/convert/_pywrap_profiler_plugin.so,sha256=oDfO8d3hD_9xtOeJOiMnkDXjXjFl8WGDLdydgdczPtM,98978504
 tensorboard_plugin_profile/convert/dcn_collective_stats_proto_to_gviz.py,sha256=o08xD3IfVOEqrtpywJA1PRTjdAKdi2TwItUY5y1keAI,4453
 tensorboard_plugin_profile/convert/diagnostics.py,sha256=Q7hKP60W5LLDiGBxefmKw9i2nZIFjOOHskQN4SCslsI,1515
 tensorboard_plugin_profile/convert/hlo_stats_proto_to_gviz.py,sha256=C3_NE7WsM8tIGv_F6RVRkwjBE2LuC5ByS4MQjUTZMtc,5434
 tensorboard_plugin_profile/convert/inference_stats_proto_to_gviz.py,sha256=28iUFcM-mslfRNMgR-G8U1-d-8fGROFzJbROJjK8u7A,11041
-tensorboard_plugin_profile/convert/input_pipeline_proto_to_gviz.py,sha256=fiD_eJQKPfLmVC9fZWF9B8GOpWNCH94LenH6Jh1P2gU,20981
+tensorboard_plugin_profile/convert/input_pipeline_proto_to_gviz.py,sha256=eBBLCCk01RSAr71uj8zw7Eb13naLwXbiDAFbKwUxgR8,21171
 tensorboard_plugin_profile/convert/kernel_stats_proto_to_gviz.py,sha256=LChKWH8qfX0MoKGcb9XGCTKz7CbUnMEnJbIN6K_fBnQ,3851
-tensorboard_plugin_profile/convert/overview_page_proto_to_gviz.py,sha256=HoXLT6AXAPW1Nq9PLXixaPng3QJnqW4MEaNAxWa47qs,10513
+tensorboard_plugin_profile/convert/overview_page_proto_to_gviz.py,sha256=KbFguCXII4RrxPq35eUNchQcTaIi10XfeyJK_B2luf4,11038
 tensorboard_plugin_profile/convert/raw_to_tool_data.py,sha256=qf4v0hKEso55stsjUT9PkvSQc0pyQ6o1x-LbM26Vov4,9598
 tensorboard_plugin_profile/convert/roofline_model_proto_to_gviz.py,sha256=KnQ0NuQm4Yl0hcwcCvTzBiTD0Z07GCXcjaOOqUE_9EU,14131
 tensorboard_plugin_profile/convert/tf_data_stats_proto_to_gviz.py,sha256=UFAjx3VBnCchxHP-i1PhLrnHG75lC33N3zY9qtWkRyQ,7584
@@ -31,7 +31,7 @@ tensorboard_plugin_profile/protobuf/tf_data_stats_pb2.py,sha256=o4OFhR9EhkCB77Ke
 tensorboard_plugin_profile/protobuf/tf_stats_pb2.py,sha256=k6qYc13yCFEQn0tjNydEjL6LackyPMkjlIg6D9dOBv8,3388
 tensorboard_plugin_profile/protobuf/tpu_input_pipeline_pb2.py,sha256=rt2w2ZoF2OFTjHS4U2p5C_lU-KejKoCQsJ9mDy4xdME,4641
 tensorboard_plugin_profile/protobuf/trace_events_pb2.py,sha256=_onaGF1Oe0Nbhb4pnA7aLammZ0RxcktvWo0vFGDufpg,3007
-tensorboard_plugin_profile/static/bundle.js,sha256=iD9em6pmVfS_d8rOTWlxEBS6w56s4_k8TZ-qP1qRE_g,23248210
+tensorboard_plugin_profile/static/bundle.js,sha256=DPZTeSE7iQ9cvxg0Qqjll_EO_Cnd1HOaQ9-PnedIcGs,23243165
 tensorboard_plugin_profile/static/index.html,sha256=D7kNoKIZ3Ntog_escZOXiDQOdY26-yZl5WLielNfrbo,1741
 tensorboard_plugin_profile/static/index.js,sha256=6k9xoStOyq8WLqAbeujc6RnudtedUiKSHgwGNg9_yPc,76
 tensorboard_plugin_profile/static/materialicons.woff2,sha256=RRfwo4kyIt8HMUExPBeMy8mTQ_OQP7EgIxc7DZ3nirk,60832
@@ -39,8 +39,8 @@ tensorboard_plugin_profile/static/styles.css,sha256=ZHVyDmih7JYJrs8kIXEdpqL-HreI
 tensorboard_plugin_profile/static/trace_viewer_index.html,sha256=LmV8aRsWN5dsZ38Y34duO7X6Rp_Qx8AWH7800VzyM8Q,136138
 tensorboard_plugin_profile/static/trace_viewer_index.js,sha256=UbduBE_PYghyLaEIJf73H12zcfSzq18kqa4em3LRLjI,3496391
 tensorboard_plugin_profile/static/zone.js,sha256=YMw-4VujXc6xhxsSyJ15ImBvBq2aKPlsCy5v82GTk20,572373
-tbp_nightly-2.20.0a20250203.dist-info/METADATA,sha256=u697rG6UGyjNFSNOyWGkWxURugxJ5Ho-ZKGqWHpHXXY,4985
-tbp_nightly-2.20.0a20250203.dist-info/WHEEL,sha256=fYnQWxCHvBJNxhGkg7OatEAocHdlLlyd_sulNRAgAtw,111
-tbp_nightly-2.20.0a20250203.dist-info/entry_points.txt,sha256=6jKbqx68mperaU0gjUeI7foBJwSl2tslwKII3q92hmA,101
-tbp_nightly-2.20.0a20250203.dist-info/top_level.txt,sha256=lhPzC2a84VfVAbBlTgLUSZwuuUl9uTj5R1zZMylY1g8,27
-tbp_nightly-2.20.0a20250203.dist-info/RECORD,,
+tbp_nightly-2.20.0a20250205.dist-info/METADATA,sha256=whrLH1GRoZ8S-mnYtYVOi03OKeiNQwYvbVjToq6H74U,4985
+tbp_nightly-2.20.0a20250205.dist-info/WHEEL,sha256=fYnQWxCHvBJNxhGkg7OatEAocHdlLlyd_sulNRAgAtw,111
+tbp_nightly-2.20.0a20250205.dist-info/entry_points.txt,sha256=6jKbqx68mperaU0gjUeI7foBJwSl2tslwKII3q92hmA,101
+tbp_nightly-2.20.0a20250205.dist-info/top_level.txt,sha256=lhPzC2a84VfVAbBlTgLUSZwuuUl9uTj5R1zZMylY1g8,27
+tbp_nightly-2.20.0a20250205.dist-info/RECORD,,

tensorboard_plugin_profile/convert/input_pipeline_proto_to_gviz.py

@@ -104,7 +104,10 @@ def get_step_breakdown_table_args_for_tpu(ipa):
   data = []
   for step_details in ipa.step_details:
     details = tpu_input_pipeline_pb2.PerTpuStepDetails()
-    step_details.Unpack(details)
+    if not step_details.Unpack(details):
+      warnings.warn("Could not unpack to PerTpuStepDetails")
+      continue
+
     tooltip = (
         "step {}: \nTime waiting for input data = {:.3f} ms, Step time ="
         " {:.3f} ms".format(
@@ -501,13 +504,20 @@ def generate_step_breakdown_table_for_tpu(ipa):
   return gviz_api.DataTable(table_description, data, custom_properties)
 
 
-def generate_step_breakdown_table(ipa):
+def generate_step_breakdown_table_for_generic(ipa):
   (table_description, data, custom_properties) = get_step_breakdown_table_args(
       ipa
   )
   return gviz_api.DataTable(table_description, data, custom_properties)
 
 
+def generate_step_breakdown_table(ipa):
+  if ipa.tag:
+    return generate_step_breakdown_table_for_tpu(ipa)
+  else:
+    return generate_step_breakdown_table_for_generic(ipa)
+
+
 def generate_input_op_table(ipa):
   (table_description, data, custom_properties) = get_input_op_table_args(ipa)
   return gviz_api.DataTable(table_description, data, custom_properties)
@@ -521,15 +531,12 @@ def generate_recommendation_table(ipa):
 
 def generate_all_chart_tables(ipa):
   """Generates a list of gviz tables from InputPipelineAnalysisResult."""
-
   tables = []
   if ipa.tag:
-    tables.append(generate_step_breakdown_table_for_tpu(ipa))
     tables.append(generate_max_infeed_core_table(ipa))
-  else:
-    tables.append(generate_step_breakdown_table(ipa))
 
   return tables + [
+      generate_step_breakdown_table(ipa),
       generate_input_op_table(ipa),
       generate_recommendation_table(ipa),
       diag.generate_diagnostics_table(ipa.diagnostics),

tensorboard_plugin_profile/convert/overview_page_proto_to_gviz.py

@@ -180,23 +180,32 @@ def get_recommendation_table_args(overview_page_recommendation):
   table_description = [
       ("tip_type", "string", "tip_type"),
       ("link", "string", "link"),
+      ("description", "string", "description"),
   ]
 
   data = []
   for faq_tip in overview_page_recommendation.faq_tips:
-    data.append(["faq", faq_tip.link])
+    data.append(["faq", faq_tip.link, "Tool troubleshooting / FAQ"])
 
   for host_tip in overview_page_recommendation.host_tips:
-    data.append(["host", host_tip.link])
+    data.append([
+        "host",
+        host_tip.link,
+        "Next steps for reducing the Host time",
+    ])
 
   for device_tip in overview_page_recommendation.device_tips:
-    data.append(["device", device_tip.link])
+    data.append(
+        ["device", device_tip.link, "Next steps for reducing the Device time"]
+    )
 
   for doc_tip in overview_page_recommendation.documentation_tips:
-    data.append(["doc", doc_tip.link])
+    data.append(["doc", doc_tip.link, "Other useful resources"])
 
   for inference_tip in overview_page_recommendation.inference_tips:
-    data.append(["inference", inference_tip.link])
+    data.append(
+        ["inference", inference_tip.link, "Recommendations for inference run"]
+    )
 
   bottleneck = overview_page_recommendation.bottleneck
   statement = overview_page_recommendation.statement
@@ -223,6 +232,11 @@ def get_recommendation_table_args(overview_page_recommendation):
       "precision_statement": precision_statement,
   }
 
+  # Prop used for data filtering in the frontend.
+  if bottleneck in ["host", "device"]:
+    non_bottleneck_tip_type = "device" if bottleneck == "host" else "host"
+    custom_properties["non_bottleneck_tip_types"] = non_bottleneck_tip_type
+
   return (table_description, data, custom_properties)