tensorboard-plugin-profile 2.19.2__tar.gz → 2.19.3__tar.gz

{tensorboard_plugin_profile-2.19.2 → tensorboard_plugin_profile-2.19.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensorboard_plugin_profile
-Version: 2.19.2
+Version: 2.19.3
 Summary: Profile Tensorboard Plugin
 Home-page: https://github.com/openxla/xprof
 Author: Google Inc.

{tensorboard_plugin_profile-2.19.2 → tensorboard_plugin_profile-2.19.3}/tensorboard_plugin_profile/convert/roofline_model_proto_to_gviz.py

@@ -152,17 +152,17 @@ def get_roofline_model_table_args_for_gpu(roofline_model_db):
     data.append(row)
   custom_properties = {
       "device_type": roofline_model_db.device_type,
-      "has_cmem": roofline_model_db.has_cmem,
-      "has_merged_vmem": roofline_model_db.has_merged_vmem,
-      "peak_flop_rate": roofline_model_db.peak_flop_rate,
-      "peak_hbm_bw": roofline_model_db.peak_hbm_bw,
-      "peak_shml1_write_bw": roofline_model_db.peak_vmem_write_bw,
-      "hbm_ridge_point": ridge_point(
+      "has_cmem": str(int(roofline_model_db.has_cmem)),
+      "has_merged_vmem": str(int(roofline_model_db.has_merged_vmem)),
+      "peak_flop_rate": str(roofline_model_db.peak_flop_rate),
+      "peak_hbm_bw": str(roofline_model_db.peak_hbm_bw),
+      "peak_shml1_write_bw": str(roofline_model_db.peak_vmem_write_bw),
+      "hbm_ridge_point": str(ridge_point(
           roofline_model_db.peak_flop_rate, roofline_model_db.peak_hbm_bw
-      ),
-      "shml1_write_ridge_point": ridge_point(
+      )),
+      "shml1_write_ridge_point": str(ridge_point(
           roofline_model_db.peak_flop_rate, roofline_model_db.peak_vmem_write_bw
-      ),
+      )),
   }
   return (table_description, data, custom_properties)
 

{tensorboard_plugin_profile-2.19.2 → tensorboard_plugin_profile-2.19.3}/tensorboard_plugin_profile/profile_plugin.py

@@ -731,7 +731,7 @@ class ProfilePlugin(base_plugin.TBPlugin):
       return respond(
           {'error': 'TensorFlow is not installed.'},
           'application/json',
-          code=200,
+          code=500,
       )
 
     def get_worker_list(
@@ -771,12 +771,12 @@ class ProfilePlugin(base_plugin.TBPlugin):
         )
         master_grpc_addr = tpu_cluster_resolver.get_master()
       except (ImportError, RuntimeError) as err:
-        return respond({'error': repr(err)}, 'application/json', code=200)
+        return respond({'error': repr(err)}, 'application/json', code=500)
       except (ValueError, TypeError):
         return respond(
             {'error': 'no TPUs with the specified names exist.'},
             'application/json',
-            code=200,
+            code=500,
         )
       if not worker_list:
         worker_list = get_worker_list(tpu_cluster_resolver)
@@ -817,7 +817,7 @@ class ProfilePlugin(base_plugin.TBPlugin):
       return respond(
           {'error': str(e)},
           'application/json',
-          code=200,
+          code=500,
       )
 
   def _get_graph_viewer_options(