xprof-nightly 2.22.3a20251208__cp311-none-manylinux2014_x86_64.whl

xprof/protobuf/memory_viewer_preprocess_pb2.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/memory_viewer_preprocess.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/memory_viewer_preprocess.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import source_info_pb2 as plugin_dot_xprof_dot_protobuf_dot_source__info__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4plugin/xprof/protobuf/memory_viewer_preprocess.proto\x12\x13tensorflow.profiler\x1a\'plugin/xprof/protobuf/source_info.proto\"\xc0\x02\n\nHeapObject\x12\x12\n\x08numbered\x18\x01 \x01(\x05H\x00\x12\x0f\n\x05named\x18\x02 \x01(\tH\x00\x12\r\n\x05label\x18\x03 \x01(\t\x12\x19\n\x11logical_buffer_id\x18\x04 \x01(\x05\x12\x1f\n\x17logical_buffer_size_mib\x18\x05 \x01(\x01\x12\x1a\n\x12unpadded_shape_mib\x18\x06 \x01(\x01\x12\x18\n\x10instruction_name\x18\x07 \x01(\t\x12\x14\n\x0cshape_string\x18\x08 \x01(\t\x12\x12\n\ntf_op_name\x18\t \x01(\t\x12\x12\n\ngroup_name\x18\n \x01(\t\x12\x0f\n\x07op_code\x18\x0b \x01(\t\x12\x34\n\x0bsource_info\x18\x0c \x01(\x0b\x32\x1f.tensorflow.profiler.SourceInfoB\x07\n\x05\x63olor\"*\n\nBufferSpan\x12\r\n\x05start\x18\x01 \x01(\x05\x12\r\n\x05limit\x18\x02 \x01(\x05\"c\n\rLogicalBuffer\x12\n\n\x02id\x18\x01 \x01(\x03\x12\r\n\x05shape\x18\x02 \x01(\t\x12\x10\n\x08size_mib\x18\x03 \x01(\x01\x12\x10\n\x08hlo_name\x18\x04 \x01(\t\x12\x13\n\x0bshape_index\x18\x05 \x03(\x03\"\x97\x01\n\x10\x42ufferAllocation\x12\n\n\x02id\x18\x01 \x01(\x03\x12\x10\n\x08size_mib\x18\x02 \x01(\x01\x12\x12\n\nattributes\x18\x03 \x03(\t\x12;\n\x0flogical_buffers\x18\x04 \x03(\x0b\x32\".tensorflow.profiler.LogicalBuffer\x12\x14\n\x0c\x63ommon_shape\x18\x05 \x01(\t\"\xe1\x06\n\x10PreprocessResult\x12\x12\n\nheap_sizes\x18\x01 \x03(\x01\x12\x1b\n\x13unpadded_heap_sizes\x18\x02 \x03(\x01\x12\x1d\n\x15hlo_instruction_names\x18\x14 \x03(\t\x12\x31\n\x08max_heap\x18\x03 \x03(\x0b\x32\x1f.tensorflow.profiler.HeapObject\x12\x39\n\x10max_heap_by_size\x18\x04 \x03(\x0b\x32\x1f.tensorflow.profiler.HeapObject\x12[\n\x14logical_buffer_spans\x18\x05 \x03(\x0b\x32=.tensorflow.profiler.PreprocessResult.LogicalBufferSpansEntry\x12\x1b\n\x13max_heap_to_by_size\x18\x06 \x03(\x05\x12\x1b\n\x13\x62y_size_to_max_heap\x18\x07 \x03(\x05\x12\x13\n\x0bmodule_name\x18\x08 \x01(\t\x12\x1e\n\x16\x65ntry_computation_name\x18\t \x01(\t\x12\x15\n\rpeak_heap_mib\x18\n \x01(\x01\x12\x1e\n\x16peak_unpadded_heap_mib\x18\x0b \x01(\x01\x12\x1f\n\x17peak_heap_size_position\x18\x0c \x01(\x05\x12(\n entry_computation_parameters_mib\x18\r \x01(\x01\x12\x18\n\x10non_reusable_mib\x18\x0e \x01(\x01\x12\x1a\n\x12maybe_live_out_mib\x18\x0f \x01(\x01\x12#\n\x1btotal_buffer_allocation_mib\x18\x12 \x01(\x01\x12(\n indefinite_buffer_allocation_mib\x18\x13 \x01(\x01\x12\x43\n\x14indefinite_lifetimes\x18\x10 \x03(\x0b\x32%.tensorflow.profiler.BufferAllocation\x12\x1b\n\x13\x61llocation_timeline\x18\x11 \x01(\t\x1aZ\n\x17LogicalBufferSpansEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12.\n\x05value\x18\x02 \x01(\x0b\x32\x1f.tensorflow.profiler.BufferSpan:\x02\x38\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.memory_viewer_preprocess_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_PREPROCESSRESULT_LOGICALBUFFERSPANSENTRY']._loaded_options = None
+  _globals['_PREPROCESSRESULT_LOGICALBUFFERSPANSENTRY']._serialized_options = b'8\001'
+  _globals['_HEAPOBJECT']._serialized_start=119
+  _globals['_HEAPOBJECT']._serialized_end=439
+  _globals['_BUFFERSPAN']._serialized_start=441
+  _globals['_BUFFERSPAN']._serialized_end=483
+  _globals['_LOGICALBUFFER']._serialized_start=485
+  _globals['_LOGICALBUFFER']._serialized_end=584
+  _globals['_BUFFERALLOCATION']._serialized_start=587
+  _globals['_BUFFERALLOCATION']._serialized_end=738
+  _globals['_PREPROCESSRESULT']._serialized_start=741
+  _globals['_PREPROCESSRESULT']._serialized_end=1606
+  _globals['_PREPROCESSRESULT_LOGICALBUFFERSPANSENTRY']._serialized_start=1516
+  _globals['_PREPROCESSRESULT_LOGICALBUFFERSPANSENTRY']._serialized_end=1606
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/op_metrics_pb2.py

@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/op_metrics.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/op_metrics.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import source_info_pb2 as plugin_dot_xprof_dot_protobuf_dot_source__info__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&plugin/xprof/protobuf/op_metrics.proto\x12\x13tensorflow.profiler\x1a\'plugin/xprof/protobuf/source_info.proto\"\xc8\x03\n\x0fPerformanceInfo\x12\x12\n\x05\x66lops\x18\x01 \x01(\x03H\x00\x88\x01\x01\x12\x1b\n\x0e\x62ytes_accessed\x18\x02 \x01(\x03H\x01\x88\x01\x01\x12V\n\x19memory_accessed_breakdown\x18\x03 \x03(\x0b\x32\x33.tensorflow.profiler.PerformanceInfo.MemoryAccessed\x1a\x88\x02\n\x0eMemoryAccessed\x12\x14\n\x07is_read\x18\x01 \x01(\x08H\x00\x88\x01\x01\x12Z\n\x0cmemory_space\x18\x02 \x01(\x0e\x32?.tensorflow.profiler.PerformanceInfo.MemoryAccessed.MemorySpaceH\x01\x88\x01\x01\x12\x1b\n\x0e\x62ytes_accessed\x18\x03 \x01(\x03H\x02\x88\x01\x01\"7\n\x0bMemorySpace\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03HBM\x10\x01\x12\x08\n\x04\x43MEM\x10\x02\x12\x08\n\x04VMEM\x10\x03\x42\n\n\x08_is_readB\x0f\n\r_memory_spaceB\x11\n\x0f_bytes_accessedB\x08\n\x06_flopsB\x11\n\x0f_bytes_accessedJ\x04\x08\x04\x10\x05\"\xc3\x01\n\x0eLayoutAnalysis\x12\x41\n\ndimensions\x18\x01 \x03(\x0b\x32-.tensorflow.profiler.LayoutAnalysis.Dimension\x1an\n\tDimension\x12\x0c\n\x04size\x18\x01 \x01(\x05\x12\x11\n\talignment\x18\x02 \x01(\x05\x12@\n\tsemantics\x18\x03 \x01(\x0e\x32-.tensorflow.profiler.LayoutDimensionSemantics\"_\n\x15MemoryAccessBreakdown\x12\x46\n\x0fmemory_accessed\x18\x01 \x03(\x0b\x32-.tensorflow.profiler.OpMetrics.MemoryAccessed\"\x95\x07\n\tOpMetrics\x12\x15\n\rhlo_module_id\x18\r \x01(\x04\x12\x0c\n\x04name\x18\x06 \x01(\t\x12\x11\n\tlong_name\x18\x14 \x01(\t\x12\x10\n\x08\x63\x61tegory\x18\x0b \x01(\t\x12\x12\n\nprovenance\x18\x0c \x01(\t\x12\x10\n\x08is_eager\x18\x12 \x01(\x08\x12\x13\n\x0boccurrences\x18\x03 \x01(\r\x12\x0f\n\x07time_ps\x18\x07 \x01(\x04\x12\x1a\n\x12normalized_time_ps\x18\x1b \x01(\x04\x12\x13\n\x0bmin_time_ps\x18\x11 \x01(\x04\x12\x14\n\x0cself_time_ps\x18\x01 \x01(\x04\x12\r\n\x05\x66lops\x18\x02 \x01(\x04\x12\x13\n\x0bmodel_flops\x18\x18 \x01(\x04\x12\x13\n\x0b\x66ingerprint\x18\x19 \x01(\x04\x12\x16\n\x0e\x62ytes_accessed\x18\x05 \x01(\x04\x12P\n\x19memory_accessed_breakdown\x18\x13 \x03(\x0b\x32-.tensorflow.profiler.OpMetrics.MemoryAccessed\x12\x14\n\x0c\x64ma_stall_ps\x18\n \x01(\x04\x12\x33\n\x06layout\x18\x0e \x01(\x0b\x32#.tensorflow.profiler.LayoutAnalysis\x12\x19\n\x11\x64\x65\x64uplicated_name\x18\x0f \x01(\t\x12\x32\n\x08\x63hildren\x18\x10 \x01(\x0b\x32 .tensorflow.profiler.OpMetricsDb\x12\x11\n\tnum_cores\x18\x15 \x01(\r\x12\"\n\x1a\x63omputation_primitive_size\x18\x16 \x01(\r\x12\x11\n\tautotuned\x18\x17 \x01(\x08\x12\x39\n\x0bsource_info\x18\x1a \x01(\x0b\x32\x1f.tensorflow.profiler.SourceInfoH\x00\x88\x01\x01\x1a\xc6\x01\n\x0eMemoryAccessed\x12S\n\x0eoperation_type\x18\x01 \x01(\x0e\x32;.tensorflow.profiler.OpMetrics.MemoryAccessed.OperationType\x12\x14\n\x0cmemory_space\x18\x02 \x01(\x04\x12\x16\n\x0e\x62ytes_accessed\x18\x03 \x01(\x04\"1\n\rOperationType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x08\n\x04READ\x10\x01\x12\t\n\x05WRITE\x10\x02\x42\x0e\n\x0c_source_infoJ\x04\x08\x04\x10\x05J\x04\x08\x08\x10\tJ\x04\x08\t\x10\n\"D\n\x0ePrecisionStats\x12\x18\n\x10\x63ompute_16bit_ps\x18\x01 \x01(\x04\x12\x18\n\x10\x63ompute_32bit_ps\x18\x02 \x01(\x04\"\x8d\x03\n\x0bOpMetricsDb\x12\x32\n\nmetrics_db\x18\n \x03(\x0b\x32\x1e.tensorflow.profiler.OpMetrics\x12)\n!total_host_infeed_enq_duration_ps\x18\x02 \x01(\x04\x12\x35\n-total_host_infeed_enq_start_timestamp_ps_diff\x18\x03 \x01(\x04\x12\x15\n\rtotal_time_ps\x18\x0b \x01(\x04\x12\x18\n\x10total_op_time_ps\x18\x0c \x01(\x04\x12#\n\x1bnormalized_total_op_time_ps\x18\x10 \x01(\x04\x12<\n\x0fprecision_stats\x18\r \x01(\x0b\x32#.tensorflow.profiler.PrecisionStats\x12\x14\n\x0cidle_time_ps\x18\x0e \x01(\x04\x12\x14\n\x0c\x62usy_time_ps\x18\x0f \x01(\x04J\x04\x08\x01\x10\x02J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\n*\xe9\x01\n\tMemBwType\x12\x15\n\x11MEM_BW_TYPE_FIRST\x10\x00\x12\x16\n\x12MEM_BW_TYPE_HBM_RW\x10\x00\x12\x17\n\x13MEM_BW_TYPE_SRAM_RD\x10\x01\x12\x17\n\x13MEM_BW_TYPE_SRAM_WR\x10\x02\x12\x17\n\x13MEM_BW_TYPE_CMEM_RD\x10\x03\x12\x17\n\x13MEM_BW_TYPE_CMEM_WR\x10\x04\x12\x17\n\x13MEM_BW_TYPE_VMEM_RD\x10\x05\x12\x17\n\x13MEM_BW_TYPE_VMEM_WR\x10\x06\x12\x13\n\x0fMEM_BW_TYPE_MAX\x10\x02\x1a\x02\x10\x01*w\n\x0bMemorySpace\x12\x1a\n\x16MEMORY_SPACE_UNDEFINED\x10\x00\x12\x14\n\x10MEMORY_SPACE_HBM\x10\x01\x12\x1c\n\x14MEMORY_SPACE_ON_CHIP\x10\xfe\xff\xff\xff\x07\x12\x18\n\x10MEMORY_SPACE_ALL\x10\xff\xff\xff\xff\x07*V\n\x18LayoutDimensionSemantics\x12\x15\n\x11UNKNOWN_SEMANTICS\x10\x00\x12\x0b\n\x07\x46\x45\x41TURE\x10\x01\x12\t\n\x05\x42\x41TCH\x10\x02\x12\x0b\n\x07SPATIAL\x10\x03\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.op_metrics_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_MEMBWTYPE']._loaded_options = None
+  _globals['_MEMBWTYPE']._serialized_options = b'\020\001'
+  _globals['_MEMBWTYPE']._serialized_start=2249
+  _globals['_MEMBWTYPE']._serialized_end=2482
+  _globals['_MEMORYSPACE']._serialized_start=2484
+  _globals['_MEMORYSPACE']._serialized_end=2603
+  _globals['_LAYOUTDIMENSIONSEMANTICS']._serialized_start=2605
+  _globals['_LAYOUTDIMENSIONSEMANTICS']._serialized_end=2691
+  _globals['_PERFORMANCEINFO']._serialized_start=105
+  _globals['_PERFORMANCEINFO']._serialized_end=561
+  _globals['_PERFORMANCEINFO_MEMORYACCESSED']._serialized_start=262
+  _globals['_PERFORMANCEINFO_MEMORYACCESSED']._serialized_end=526
+  _globals['_PERFORMANCEINFO_MEMORYACCESSED_MEMORYSPACE']._serialized_start=423
+  _globals['_PERFORMANCEINFO_MEMORYACCESSED_MEMORYSPACE']._serialized_end=478
+  _globals['_LAYOUTANALYSIS']._serialized_start=564
+  _globals['_LAYOUTANALYSIS']._serialized_end=759
+  _globals['_LAYOUTANALYSIS_DIMENSION']._serialized_start=649
+  _globals['_LAYOUTANALYSIS_DIMENSION']._serialized_end=759
+  _globals['_MEMORYACCESSBREAKDOWN']._serialized_start=761
+  _globals['_MEMORYACCESSBREAKDOWN']._serialized_end=856
+  _globals['_OPMETRICS']._serialized_start=859
+  _globals['_OPMETRICS']._serialized_end=1776
+  _globals['_OPMETRICS_MEMORYACCESSED']._serialized_start=1544
+  _globals['_OPMETRICS_MEMORYACCESSED']._serialized_end=1742
+  _globals['_OPMETRICS_MEMORYACCESSED_OPERATIONTYPE']._serialized_start=1693
+  _globals['_OPMETRICS_MEMORYACCESSED_OPERATIONTYPE']._serialized_end=1742
+  _globals['_PRECISIONSTATS']._serialized_start=1778
+  _globals['_PRECISIONSTATS']._serialized_end=1846
+  _globals['_OPMETRICSDB']._serialized_start=1849
+  _globals['_OPMETRICSDB']._serialized_end=2246
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/op_profile_pb2.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/op_profile.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/op_profile.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import source_info_pb2 as plugin_dot_xprof_dot_protobuf_dot_source__info__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&plugin/xprof/protobuf/op_profile.proto\x12\x1etensorflow.profiler.op_profile\x1a\'plugin/xprof/protobuf/source_info.proto\"\x80\x04\n\x07Profile\x12\x39\n\x0b\x62y_category\x18\x01 \x01(\x0b\x32$.tensorflow.profiler.op_profile.Node\x12\x38\n\nby_program\x18\x04 \x01(\x0b\x32$.tensorflow.profiler.op_profile.Node\x12\x13\n\x0b\x64\x65vice_type\x18\x05 \x01(\t\x12&\n\x1e\x61gg_dvfs_time_scale_multiplier\x18\n \x01(\x01\x12\x46\n\x18\x62y_category_exclude_idle\x18\x06 \x01(\x0b\x32$.tensorflow.profiler.op_profile.Node\x12\x45\n\x17\x62y_program_exclude_idle\x18\x07 \x01(\x0b\x32$.tensorflow.profiler.op_profile.Node\x12;\n\rby_provenance\x18\x08 \x01(\x0b\x32$.tensorflow.profiler.op_profile.Node\x12H\n\x1a\x62y_provenance_exclude_idle\x18\t \x01(\x0b\x32$.tensorflow.profiler.op_profile.NodeJ\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04R\x14\x62y_program_structureR\x0bper_program\"\x8a\x07\n\x04Node\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x38\n\x07metrics\x18\x02 \x01(\x0b\x32\'.tensorflow.profiler.op_profile.Metrics\x12\x36\n\x08\x63hildren\x18\x03 \x03(\x0b\x32$.tensorflow.profiler.op_profile.Node\x12L\n\x08\x63\x61tegory\x18\x04 \x01(\x0b\x32\x38.tensorflow.profiler.op_profile.Node.InstructionCategoryH\x00\x12\x42\n\x03xla\x18\x05 \x01(\x0b\x32\x33.tensorflow.profiler.op_profile.Node.XLAInstructionH\x00\x12\x14\n\x0cnum_children\x18\x06 \x01(\x05\x1a\x15\n\x13InstructionCategory\x1a\xb6\x04\n\x0eXLAInstruction\x12\n\n\x02op\x18\x01 \x01(\t\x12\x12\n\nexpression\x18\x02 \x01(\t\x12\x12\n\nprovenance\x18\x03 \x01(\t\x12\x10\n\x08\x63\x61tegory\x18\x04 \x01(\t\x12R\n\x06layout\x18\x05 \x01(\x0b\x32\x42.tensorflow.profiler.op_profile.Node.XLAInstruction.LayoutAnalysis\x12\"\n\x1a\x63omputation_primitive_size\x18\x06 \x01(\r\x12\x13\n\x0b\x66ingerprint\x18\x07 \x01(\x04\x12\x12\n\nprogram_id\x18\x08 \x01(\x04\x12\x39\n\x0bsource_info\x18\t \x01(\x0b\x32\x1f.tensorflow.profiler.SourceInfoH\x00\x88\x01\x01\x12\"\n\x15xprof_kernel_metadata\x18\n \x01(\tH\x01\x88\x01\x01\x1a\xb3\x01\n\x0eLayoutAnalysis\x12`\n\ndimensions\x18\x01 \x03(\x0b\x32L.tensorflow.profiler.op_profile.Node.XLAInstruction.LayoutAnalysis.Dimension\x1a?\n\tDimension\x12\x0c\n\x04size\x18\x01 \x01(\x05\x12\x11\n\talignment\x18\x02 \x01(\x05\x12\x11\n\tsemantics\x18\x03 \x01(\tB\x0e\n\x0c_source_infoB\x18\n\x16_xprof_kernel_metadataB\n\n\x08\x63ontents\"\x88\x02\n\x07Metrics\x12\r\n\x05\x66lops\x18\x02 \x01(\x01\x12\x16\n\x0euncapped_flops\x18\x13 \x01(\x01\x12\x17\n\x0f\x62\x61ndwidth_utils\x18\x05 \x03(\x01\x12\x10\n\x08raw_time\x18\x0b \x01(\x01\x12\x11\n\traw_flops\x18\x0c \x01(\x01\x12\x12\n\nbf16_flops\x18\x12 \x01(\x01\x12\x1a\n\x12normalized_time_ps\x18\x14 \x01(\x01\x12 \n\x18raw_bytes_accessed_array\x18\x0f \x03(\x01\x12\x13\n\x0boccurrences\x18\x10 \x01(\r\x12\x13\n\x0b\x61vg_time_ps\x18\x11 \x01(\x01J\x04\x08\x01\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\r\x10\x0eJ\x04\x08\x0e\x10\x0f\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.op_profile_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_PROFILE']._serialized_start=116
+  _globals['_PROFILE']._serialized_end=628
+  _globals['_NODE']._serialized_start=631
+  _globals['_NODE']._serialized_end=1537
+  _globals['_NODE_INSTRUCTIONCATEGORY']._serialized_start=935
+  _globals['_NODE_INSTRUCTIONCATEGORY']._serialized_end=956
+  _globals['_NODE_XLAINSTRUCTION']._serialized_start=959
+  _globals['_NODE_XLAINSTRUCTION']._serialized_end=1525
+  _globals['_NODE_XLAINSTRUCTION_LAYOUTANALYSIS']._serialized_start=1304
+  _globals['_NODE_XLAINSTRUCTION_LAYOUTANALYSIS']._serialized_end=1483
+  _globals['_NODE_XLAINSTRUCTION_LAYOUTANALYSIS_DIMENSION']._serialized_start=1420
+  _globals['_NODE_XLAINSTRUCTION_LAYOUTANALYSIS_DIMENSION']._serialized_end=1483
+  _globals['_METRICS']._serialized_start=1540
+  _globals['_METRICS']._serialized_end=1804
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/op_stats_pb2.py

@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/op_stats.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/op_stats.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import diagnostics_pb2 as plugin_dot_xprof_dot_protobuf_dot_diagnostics__pb2
+from xprof.protobuf import hardware_types_pb2 as plugin_dot_xprof_dot_protobuf_dot_hardware__types__pb2
+from xprof.protobuf import kernel_stats_pb2 as plugin_dot_xprof_dot_protobuf_dot_kernel__stats__pb2
+from xprof.protobuf import op_metrics_pb2 as plugin_dot_xprof_dot_protobuf_dot_op__metrics__pb2
+from xprof.protobuf import power_metrics_pb2 as plugin_dot_xprof_dot_protobuf_dot_power__metrics__pb2
+from xprof.protobuf import source_stats_pb2 as plugin_dot_xprof_dot_protobuf_dot_source__stats__pb2
+from xprof.protobuf import steps_db_pb2 as plugin_dot_xprof_dot_protobuf_dot_steps__db__pb2
+from xprof.protobuf import tf_function_pb2 as plugin_dot_xprof_dot_protobuf_dot_tf__function__pb2
+from xprof.protobuf import topology_pb2 as plugin_dot_xprof_dot_protobuf_dot_topology__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$plugin/xprof/protobuf/op_stats.proto\x12\x13tensorflow.profiler\x1a\'plugin/xprof/protobuf/diagnostics.proto\x1a*plugin/xprof/protobuf/hardware_types.proto\x1a(plugin/xprof/protobuf/kernel_stats.proto\x1a&plugin/xprof/protobuf/op_metrics.proto\x1a)plugin/xprof/protobuf/power_metrics.proto\x1a(plugin/xprof/protobuf/source_stats.proto\x1a$plugin/xprof/protobuf/steps_db.proto\x1a\'plugin/xprof/protobuf/tf_function.proto\x1a$plugin/xprof/protobuf/topology.proto\"\xfd\x01\n\x07PerfEnv\x12\"\n\x1apeak_tera_flops_per_second\x18\x01 \x01(\x01\x12%\n\x1dpeak_bw_giga_bytes_per_second\x18\x04 \x01(\x01\x12)\n!peak_hbm_bw_giga_bytes_per_second\x18\x02 \x01(\x01\x12&\n\x1epeak_bws_giga_bytes_per_second\x18\x05 \x03(\x01\x12\x13\n\x0bridge_point\x18\x03 \x01(\x01\x12\x10\n\x08has_cmem\x18\x06 \x01(\x08\x12\x17\n\x0fhas_merged_vmem\x18\x07 \x01(\x08\x12\x14\n\x0chas_megacore\x18\x08 \x01(\x08\"z\n\x1cHostIndependentJobInfoResult\x12\x13\n\x0b\x63hange_list\x18\x01 \x01(\x03\x12\x12\n\nbuild_time\x18\x02 \x01(\x03\x12\x14\n\x0c\x62uild_target\x18\x03 \x01(\t\x12\x1b\n\x13profile_duration_ms\x18\x04 \x01(\r\"\x85\x01\n\x1aHostDependentJobInfoResult\x12\x0f\n\x07host_id\x18\x01 \x01(\t\x12\x14\n\x0c\x63ommand_line\x18\x02 \x01(\t\x12\x12\n\nstart_time\x18\x03 \x01(\x03\x12\x13\n\x0b\x62ns_address\x18\x04 \x01(\t\x12\x17\n\x0fprofile_time_ns\x18\x05 \x01(\x04\"s\n\x0eSystemTopology\x12\x13\n\x0bx_dimension\x18\x01 \x01(\x03\x12\x13\n\x0by_dimension\x18\x02 \x01(\x03\x12\x13\n\x0bz_dimension\x18\x03 \x01(\x03\x12\"\n\x1anum_expected_reduced_chips\x18\x04 \x01(\x03\"\xa6\x05\n\x0eRunEnvironment\x12\x12\n\nhost_count\x18\x01 \x01(\x05\x12\x12\n\ntask_count\x18\x02 \x01(\x05\x12\x45\n\thostnames\x18\x03 \x03(\x0b\x32\x32.tensorflow.profiler.RunEnvironment.HostnamesEntry\x12\x13\n\x0b\x64\x65vice_type\x18\x04 \x01(\t\x12\x19\n\x11\x64\x65vice_core_count\x18\x05 \x01(\x05\x12T\n\x19host_independent_job_info\x18\x07 \x01(\x0b\x32\x31.tensorflow.profiler.HostIndependentJobInfoResult\x12P\n\x17host_dependent_job_info\x18\x08 \x03(\x0b\x32/.tensorflow.profiler.HostDependentJobInfoResult\x12\x15\n\rreplica_count\x18\t \x01(\x05\x12\x1d\n\x15num_cores_per_replica\x18\n \x01(\x05\x12\x18\n\x10host_trace_level\x18\x0c \x01(\r\x12\x36\n\x0fsystem_topology\x18\r \x01(\x0b\x32\x1d.tensorflow.profiler.Topology\x12\x13\n\x0bis_training\x18\x0e \x01(\x08\x12\x38\n\rpower_metrics\x18\x0f \x01(\x0b\x32!.tensorflow.profiler.PowerMetrics\x12\x38\n\rhardware_type\x18\x10 \x01(\x0e\x32!.tensorflow.profiler.HardwareType\x1a\x30\n\x0eHostnamesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x08:\x02\x38\x01J\x04\x08\x06\x10\x07J\x04\x08\x0b\x10\x0c\"\xa8\x01\n\x0b\x43oreDetails\x12\x10\n\x08hostname\x18\x01 \x01(\t\x12\x16\n\x0e\x64\x65vice_ordinal\x18\x02 \x01(\r\x12\x10\n\x08\x63ore_num\x18\x03 \x01(\r\x12\x15\n\rlocal_chip_id\x18\x04 \x01(\r\x12\x16\n\x0eglobal_chip_id\x18\x05 \x01(\r\x12\x16\n\x0eglobal_core_id\x18\x06 \x01(\r\x12\x16\n\x0eis_sparse_core\x18\x07 \x01(\x08\"d\n\x18PerformanceCounterResult\x12\'\n\x1fmatrix_unit_utilization_percent\x18\x01 \x01(\x01\x12\x1f\n\x17hbm_utilization_percent\x18\x02 \x01(\x01\"\xf7\x07\n\x07OpStats\x12<\n\x12host_op_metrics_db\x18\x01 \x01(\x0b\x32 .tensorflow.profiler.OpMetricsDb\x12>\n\x14\x64\x65vice_op_metrics_db\x18\x02 \x01(\x0b\x32 .tensorflow.profiler.OpMetricsDb\x12L\n\"hlo_metrics_db_complete_steps_only\x18\n \x01(\x0b\x32 .tensorflow.profiler.OpMetricsDb\x12.\n\x08perf_env\x18\x03 \x01(\x0b\x32\x1c.tensorflow.profiler.PerfEnv\x12\x38\n\x07step_db\x18\x04 \x01(\x0b\x32\'.tensorflow.profiler.StepDatabaseResult\x12<\n\x0frun_environment\x18\x05 \x01(\x0b\x32#.tensorflow.profiler.RunEnvironment\x12;\n\x0fkernel_stats_db\x18\x06 \x01(\x0b\x32\".tensorflow.profiler.KernelStatsDb\x12\x39\n\x0etf_function_db\x18\x08 \x01(\x0b\x32!.tensorflow.profiler.TfFunctionDb\x12M\n\x12\x63ore_id_to_details\x18\x0b \x03(\x0b\x32\x31.tensorflow.profiler.OpStats.CoreIdToDetailsEntry\x12\x35\n\x0b\x64iagnostics\x18\t \x01(\x0b\x32 .tensorflow.profiler.Diagnostics\x12T\n\x16program_id_to_name_map\x18\x0c \x03(\x0b\x32\x34.tensorflow.profiler.OpStats.ProgramIdToNameMapEntry\x12Q\n\x1aperformance_counter_result\x18\r \x01(\x0b\x32-.tensorflow.profiler.PerformanceCounterResult\x12\x36\n\x0csource_stats\x18\x0e \x01(\x0b\x32 .tensorflow.profiler.SourceStats\x1aX\n\x14\x43oreIdToDetailsEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12/\n\x05value\x18\x02 \x01(\x0b\x32 .tensorflow.profiler.CoreDetails:\x02\x38\x01\x1a\x39\n\x17ProgramIdToNameMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01J\x04\x08\x07\x10\x08\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.op_stats_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_RUNENVIRONMENT_HOSTNAMESENTRY']._loaded_options = None
+  _globals['_RUNENVIRONMENT_HOSTNAMESENTRY']._serialized_options = b'8\001'
+  _globals['_OPSTATS_COREIDTODETAILSENTRY']._loaded_options = None
+  _globals['_OPSTATS_COREIDTODETAILSENTRY']._serialized_options = b'8\001'
+  _globals['_OPSTATS_PROGRAMIDTONAMEMAPENTRY']._loaded_options = None
+  _globals['_OPSTATS_PROGRAMIDTONAMEMAPENTRY']._serialized_options = b'8\001'
+  _globals['_PERFENV']._serialized_start=431
+  _globals['_PERFENV']._serialized_end=684
+  _globals['_HOSTINDEPENDENTJOBINFORESULT']._serialized_start=686
+  _globals['_HOSTINDEPENDENTJOBINFORESULT']._serialized_end=808
+  _globals['_HOSTDEPENDENTJOBINFORESULT']._serialized_start=811
+  _globals['_HOSTDEPENDENTJOBINFORESULT']._serialized_end=944
+  _globals['_SYSTEMTOPOLOGY']._serialized_start=946
+  _globals['_SYSTEMTOPOLOGY']._serialized_end=1061
+  _globals['_RUNENVIRONMENT']._serialized_start=1064
+  _globals['_RUNENVIRONMENT']._serialized_end=1742
+  _globals['_RUNENVIRONMENT_HOSTNAMESENTRY']._serialized_start=1682
+  _globals['_RUNENVIRONMENT_HOSTNAMESENTRY']._serialized_end=1730
+  _globals['_COREDETAILS']._serialized_start=1745
+  _globals['_COREDETAILS']._serialized_end=1913
+  _globals['_PERFORMANCECOUNTERRESULT']._serialized_start=1915
+  _globals['_PERFORMANCECOUNTERRESULT']._serialized_end=2015
+  _globals['_OPSTATS']._serialized_start=2018
+  _globals['_OPSTATS']._serialized_end=3033
+  _globals['_OPSTATS_COREIDTODETAILSENTRY']._serialized_start=2880
+  _globals['_OPSTATS_COREIDTODETAILSENTRY']._serialized_end=2968
+  _globals['_OPSTATS_PROGRAMIDTONAMEMAPENTRY']._serialized_start=2970
+  _globals['_OPSTATS_PROGRAMIDTONAMEMAPENTRY']._serialized_end=3027
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/overview_page_pb2.py

@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/overview_page.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/overview_page.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
+from xprof.protobuf import diagnostics_pb2 as plugin_dot_xprof_dot_protobuf_dot_diagnostics__pb2
+from xprof.protobuf import input_pipeline_pb2 as plugin_dot_xprof_dot_protobuf_dot_input__pipeline__pb2
+from xprof.protobuf import power_metrics_pb2 as plugin_dot_xprof_dot_protobuf_dot_power__metrics__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)plugin/xprof/protobuf/overview_page.proto\x12\x13tensorflow.profiler\x1a\x19google/protobuf/any.proto\x1a\'plugin/xprof/protobuf/diagnostics.proto\x1a*plugin/xprof/protobuf/input_pipeline.proto\x1a)plugin/xprof/protobuf/power_metrics.proto\"\xc2\x01\n\x0cOverviewTfOp\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08\x63\x61tegory\x18\x02 \x01(\t\x12\x1a\n\x12self_time_fraction\x18\x03 \x01(\x01\x12 \n\x18\x63umulative_time_fraction\x18\x04 \x01(\x01\x12\x11\n\tflop_rate\x18\x05 \x01(\x01\x12!\n\x19is_op_tensorcore_eligible\x18\x06 \x01(\x08\x12\x1e\n\x16is_op_using_tensorcore\x18\x07 \x01(\x08\"\x8b\t\n\x14OverviewPageAnalysis\x12\x1f\n\x17mxu_utilization_percent\x18\x01 \x01(\x01\x12 \n\x18\x64\x65vice_idle_time_percent\x18\x02 \x01(\x01\x12\x1e\n\x16host_idle_time_percent\x18\x03 \x01(\x01\x12\x39\n\x0etop_device_ops\x18\x04 \x03(\x0b\x32!.tensorflow.profiler.OverviewTfOp\x12\x13\n\x0bremark_text\x18\x05 \x01(\t\x12\x14\n\x0cremark_color\x18\x06 \x01(\t\x12:\n2flop_rate_utilization_relative_to_roofline_percent\x18\x07 \x01(\x01\x12:\n2memory_bw_utilization_relative_to_hw_limit_percent\x18\x08 \x01(\x01\x12$\n\x1c\x64\x65vice_compute_16bit_percent\x18\t \x01(\x01\x12$\n\x1c\x64\x65vice_compute_32bit_percent\x18\n \x01(\x01\x12\x1a\n\x12host_tf_op_percent\x18\x0b \x01(\x01\x12\x1c\n\x14\x64\x65vice_tf_op_percent\x18\x0c \x01(\x01\x12\x18\n\x10host_trace_level\x18\r \x01(\r\x12\"\n\x1ahost_op_time_eager_percent\x18\x0e \x01(\x01\x12$\n\x1c\x64\x65vice_op_time_eager_percent\x18\x0f \x01(\x01\x12\x32\n*device_op_time_outside_compilation_percent\x18\x10 \x01(\x01\x12!\n\x19\x64\x65vice_duty_cycle_percent\x18\x11 \x01(\x01\x12\x1f\n\x17hbm_utilization_percent\x18\x1f \x01(\x01\x12\x1f\n\x17program_goodput_percent\x18\x12 \x01(\x01\x12\x1f\n\x17sc_step_time_ms_average\x18\x13 \x01(\x01\x12\x1d\n\x15sc_infeed_time_ms_avg\x18\x14 \x01(\x01\x12\x1e\n\x16sc_outfeed_time_ms_avg\x18\x15 \x01(\x01\x12\x1b\n\x13sc_idle_time_ms_avg\x18\x16 \x01(\x01\x12\'\n\x1f\x66w_max_vdd_core_pl1_power_watts\x18\x17 \x01(\x01\x12\'\n\x1f\x66w_max_vdd_core_pl2_power_watts\x18\x18 \x01(\x01\x12\'\n\x1f\x66w_max_vdd_core_pl3_power_watts\x18\x19 \x01(\x01\x12\'\n\x1f\x66w_max_vdd_core_pl4_power_watts\x18\x1a \x01(\x01\x12\"\n\x1a\x66w_max_hbm_pl1_power_watts\x18\x1b \x01(\x01\x12\"\n\x1a\x66w_max_hbm_pl2_power_watts\x18\x1c \x01(\x01\x12\"\n\x1a\x66w_max_hbm_pl3_power_watts\x18\x1d \x01(\x01\x12\"\n\x1a\x66w_max_hbm_pl4_power_watts\x18\x1e \x01(\x01\"\x1f\n\x0fOverviewPageTip\x12\x0c\n\x04link\x18\x01 \x01(\t\"\xff\x01\n\x15GenericRecommendation\x12 \n\x18kernel_launch_bottleneck\x18\x01 \x01(\t\x12\x1f\n\x17kernel_launch_statement\x18\x02 \x01(\t\x12\x1c\n\x14\x61ll_other_bottleneck\x18\x03 \x01(\t\x12\x1b\n\x13\x61ll_other_statement\x18\x04 \x01(\t\x12\x1b\n\x13precision_statement\x18\x05 \x01(\t\x12%\n\x1d\x64\x65vice_collectives_bottleneck\x18\x06 \x01(\t\x12$\n\x1c\x64\x65vice_collectives_statement\x18\x07 \x01(\t\"\xdf\x04\n\x1aOverviewPageRecommendation\x12\x12\n\nbottleneck\x18\x01 \x01(\t\x12\x11\n\tstatement\x18\x02 \x01(\t\x12\x38\n\ninput_tips\x18\x0b \x03(\x0b\x32$.tensorflow.profiler.OverviewPageTip\x12\x18\n\x10output_statement\x18\t \x01(\t\x12\x1c\n\x14\x65\x61ger_statement_html\x18\x0c \x01(\t\x12*\n\"outside_compilation_statement_html\x18\r \x01(\t\x12\"\n\x1atf_function_statement_html\x18\n \x01(\t\x12\x37\n\thost_tips\x18\x03 \x03(\x0b\x32$.tensorflow.profiler.OverviewPageTip\x12\x39\n\x0b\x64\x65vice_tips\x18\x04 \x03(\x0b\x32$.tensorflow.profiler.OverviewPageTip\x12@\n\x12\x64ocumentation_tips\x18\x05 \x03(\x0b\x32$.tensorflow.profiler.OverviewPageTip\x12,\n\x0erecommendation\x18\x06 \x01(\x0b\x32\x14.google.protobuf.Any\x12\x36\n\x08\x66\x61q_tips\x18\x07 \x03(\x0b\x32$.tensorflow.profiler.OverviewPageTip\x12<\n\x0einference_tips\x18\x08 \x03(\x0b\x32$.tensorflow.profiler.OverviewPageTip\"\x80\x01\n\"OverviewPageHostIndependentJobInfo\x12\x13\n\x0b\x63hange_list\x18\x01 \x01(\x03\x12\x12\n\nbuild_time\x18\x02 \x01(\x03\x12\x14\n\x0c\x62uild_target\x18\x03 \x01(\t\x12\x1b\n\x13profile_duration_ms\x18\x04 \x01(\r\"\x8b\x01\n OverviewPageHostDependentJobInfo\x12\x0f\n\x07host_id\x18\x01 \x01(\t\x12\x14\n\x0c\x63ommand_line\x18\x02 \x01(\t\x12\x12\n\nstart_time\x18\x03 \x01(\x03\x12\x13\n\x0b\x62ns_address\x18\x04 \x01(\t\x12\x17\n\x0fprofile_time_ns\x18\x05 \x01(\x04\"\xb8\x04\n\x1aOverviewPageRunEnvironment\x12\x12\n\nhost_count\x18\x01 \x01(\x05\x12\x12\n\ntask_count\x18\x02 \x01(\x05\x12Q\n\thostnames\x18\x03 \x03(\x0b\x32>.tensorflow.profiler.OverviewPageRunEnvironment.HostnamesEntry\x12\x13\n\x0b\x64\x65vice_type\x18\x04 \x01(\t\x12\x19\n\x11\x64\x65vice_core_count\x18\x05 \x01(\x05\x12Z\n\x19host_independent_job_info\x18\x07 \x01(\x0b\x32\x37.tensorflow.profiler.OverviewPageHostIndependentJobInfo\x12V\n\x17host_dependent_job_info\x18\x08 \x03(\x0b\x32\x35.tensorflow.profiler.OverviewPageHostDependentJobInfo\x12\x15\n\rreplica_count\x18\t \x01(\x05\x12\x1d\n\x15num_cores_per_replica\x18\n \x01(\x05\x12\x13\n\x0bis_training\x18\x0b \x01(\x08\x12\x38\n\rpower_metrics\x18\x0c \x01(\x0b\x32!.tensorflow.profiler.PowerMetrics\x1a\x30\n\x0eHostnamesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x08:\x02\x38\x01J\x04\x08\x06\x10\x07\"\x8a\x01\n\x18OverviewLatencyBreakdown\x12\x18\n\x10total_latency_us\x18\x01 \x01(\x01\x12\x17\n\x0fhost_latency_us\x18\x02 \x01(\x01\x12\x19\n\x11\x64\x65vice_latency_us\x18\x03 \x01(\x01\x12 \n\x18\x63ommunication_latency_us\x18\x04 \x01(\x01\"\xeb\x01\n\x18OverviewInferenceLatency\x12\x1a\n\x12percentile_numbers\x18\x01 \x03(\x01\x12I\n\x12latency_breakdowns\x18\x02 \x03(\x0b\x32-.tensorflow.profiler.OverviewLatencyBreakdown\x12\x16\n\x0emax_latency_us\x18\x03 \x01(\x01\x12\x16\n\x0emin_latency_us\x18\x04 \x01(\x01\x12 \n\x13sessions_per_second\x18\x05 \x01(\x01H\x00\x88\x01\x01\x42\x16\n\x14_sessions_per_second\"\xbb\x03\n\x0cOverviewPage\x12H\n\x0frun_environment\x18\x06 \x01(\x0b\x32/.tensorflow.profiler.OverviewPageRunEnvironment\x12H\n\x0einput_analysis\x18\x02 \x01(\x0b\x32\x30.tensorflow.profiler.InputPipelineAnalysisResult\x12;\n\x08\x61nalysis\x18\x03 \x01(\x0b\x32).tensorflow.profiler.OverviewPageAnalysis\x12G\n\x0erecommendation\x18\x04 \x01(\x0b\x32/.tensorflow.profiler.OverviewPageRecommendation\x12\x35\n\x0b\x64iagnostics\x18\x08 \x01(\x0b\x32 .tensorflow.profiler.Diagnostics\x12H\n\x11inference_latency\x18\t \x01(\x0b\x32-.tensorflow.profiler.OverviewInferenceLatencyJ\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x07\x10\x08\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.overview_page_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_OVERVIEWPAGERUNENVIRONMENT_HOSTNAMESENTRY']._loaded_options = None
+  _globals['_OVERVIEWPAGERUNENVIRONMENT_HOSTNAMESENTRY']._serialized_options = b'8\001'
+  _globals['_OVERVIEWTFOP']._serialized_start=222
+  _globals['_OVERVIEWTFOP']._serialized_end=416
+  _globals['_OVERVIEWPAGEANALYSIS']._serialized_start=419
+  _globals['_OVERVIEWPAGEANALYSIS']._serialized_end=1582
+  _globals['_OVERVIEWPAGETIP']._serialized_start=1584
+  _globals['_OVERVIEWPAGETIP']._serialized_end=1615
+  _globals['_GENERICRECOMMENDATION']._serialized_start=1618
+  _globals['_GENERICRECOMMENDATION']._serialized_end=1873
+  _globals['_OVERVIEWPAGERECOMMENDATION']._serialized_start=1876
+  _globals['_OVERVIEWPAGERECOMMENDATION']._serialized_end=2483
+  _globals['_OVERVIEWPAGEHOSTINDEPENDENTJOBINFO']._serialized_start=2486
+  _globals['_OVERVIEWPAGEHOSTINDEPENDENTJOBINFO']._serialized_end=2614
+  _globals['_OVERVIEWPAGEHOSTDEPENDENTJOBINFO']._serialized_start=2617
+  _globals['_OVERVIEWPAGEHOSTDEPENDENTJOBINFO']._serialized_end=2756
+  _globals['_OVERVIEWPAGERUNENVIRONMENT']._serialized_start=2759
+  _globals['_OVERVIEWPAGERUNENVIRONMENT']._serialized_end=3327
+  _globals['_OVERVIEWPAGERUNENVIRONMENT_HOSTNAMESENTRY']._serialized_start=3273
+  _globals['_OVERVIEWPAGERUNENVIRONMENT_HOSTNAMESENTRY']._serialized_end=3321
+  _globals['_OVERVIEWLATENCYBREAKDOWN']._serialized_start=3330
+  _globals['_OVERVIEWLATENCYBREAKDOWN']._serialized_end=3468
+  _globals['_OVERVIEWINFERENCELATENCY']._serialized_start=3471
+  _globals['_OVERVIEWINFERENCELATENCY']._serialized_end=3706
+  _globals['_OVERVIEWPAGE']._serialized_start=3709
+  _globals['_OVERVIEWPAGE']._serialized_end=4152
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/pod_stats_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/pod_stats.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/pod_stats.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import diagnostics_pb2 as plugin_dot_xprof_dot_protobuf_dot_diagnostics__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%plugin/xprof/protobuf/pod_stats.proto\x12\x13tensorflow.profiler\x1a\'plugin/xprof/protobuf/diagnostics.proto\"/\n\x13StepBreakdownEvents\x12\n\n\x02id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\"\xd7\x01\n\x10PodStatsDatabase\x12=\n\x10pod_stats_record\x18\x01 \x03(\x0b\x32#.tensorflow.profiler.PodStatsRecord\x12\x35\n\x0b\x64iagnostics\x18\x03 \x01(\x0b\x32 .tensorflow.profiler.Diagnostics\x12G\n\x15step_breakdown_events\x18\x04 \x03(\x0b\x32(.tensorflow.profiler.StepBreakdownEventsJ\x04\x08\x02\x10\x03\"\xdb\x02\n\x0ePodStatsRecord\x12\x11\n\thost_name\x18\x01 \x01(\t\x12\x0f\n\x07\x63hip_id\x18\x02 \x01(\x05\x12\x0f\n\x07node_id\x18\x03 \x01(\x05\x12\x10\n\x08step_num\x18\x04 \x01(\r\x12\x19\n\x11total_duration_us\x18\x05 \x01(\x01\x12S\n\x11step_breakdown_us\x18\x13 \x03(\x0b\x32\x38.tensorflow.profiler.PodStatsRecord.StepBreakdownUsEntry\x12\x12\n\nbottleneck\x18\x0e \x01(\t\x1a\x36\n\x14StepBreakdownUsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0bJ\x04\x08\x0b\x10\x0cJ\x04\x08\x0c\x10\rJ\x04\x08\r\x10\x0eJ\x04\x08\x0f\x10\x10J\x04\x08\x10\x10\x11J\x04\x08\x11\x10\x12J\x04\x08\x12\x10\x13\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.pod_stats_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_PODSTATSRECORD_STEPBREAKDOWNUSENTRY']._loaded_options = None
+  _globals['_PODSTATSRECORD_STEPBREAKDOWNUSENTRY']._serialized_options = b'8\001'
+  _globals['_STEPBREAKDOWNEVENTS']._serialized_start=103
+  _globals['_STEPBREAKDOWNEVENTS']._serialized_end=150
+  _globals['_PODSTATSDATABASE']._serialized_start=153
+  _globals['_PODSTATSDATABASE']._serialized_end=368
+  _globals['_PODSTATSRECORD']._serialized_start=371
+  _globals['_PODSTATSRECORD']._serialized_end=718
+  _globals['_PODSTATSRECORD_STEPBREAKDOWNUSENTRY']._serialized_start=592
+  _globals['_PODSTATSRECORD_STEPBREAKDOWNUSENTRY']._serialized_end=646
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/pod_viewer_pb2.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/pod_viewer.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/pod_viewer.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import diagnostics_pb2 as plugin_dot_xprof_dot_protobuf_dot_diagnostics__pb2
+from xprof.protobuf import pod_stats_pb2 as plugin_dot_xprof_dot_protobuf_dot_pod__stats__pb2
+from xprof.protobuf import topology_pb2 as plugin_dot_xprof_dot_protobuf_dot_topology__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&plugin/xprof/protobuf/pod_viewer.proto\x12\x13tensorflow.profiler\x1a\'plugin/xprof/protobuf/diagnostics.proto\x1a%plugin/xprof/protobuf/pod_stats.proto\x1a$plugin/xprof/protobuf/topology.proto\"#\n\x0cReplicaGroup\x12\x13\n\x0breplica_ids\x18\x01 \x03(\x03\"\xac\x01\n\x0f\x41llReduceOpInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0boccurrences\x18\x02 \x01(\r\x12\x13\n\x0b\x64uration_us\x18\x03 \x01(\x01\x12\x11\n\tdata_size\x18\x04 \x01(\x04\x12\x39\n\x0ereplica_groups\x18\x05 \x03(\x0b\x32!.tensorflow.profiler.ReplicaGroup\x12\x13\n\x0b\x64\x65scription\x18\x06 \x01(\t\"\xe1\x03\n\x0bPodStatsMap\x12\x10\n\x08step_num\x18\x01 \x01(\r\x12Q\n\x12pod_stats_per_core\x18\x02 \x03(\x0b\x32\x35.tensorflow.profiler.PodStatsMap.PodStatsPerCoreEntry\x12\x34\n\nchannel_db\x18\x03 \x03(\x0b\x32 .tensorflow.profiler.ChannelInfo\x12]\n\x19\x63ore_id_to_replica_id_map\x18\x04 \x03(\x0b\x32:.tensorflow.profiler.PodStatsMap.CoreIdToReplicaIdMapEntry\x12>\n\x10\x61ll_reduce_op_db\x18\x05 \x03(\x0b\x32$.tensorflow.profiler.AllReduceOpInfo\x1a[\n\x14PodStatsPerCoreEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\x32\n\x05value\x18\x02 \x01(\x0b\x32#.tensorflow.profiler.PodStatsRecord:\x02\x38\x01\x1a;\n\x19\x43oreIdToReplicaIdMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12\r\n\x05value\x18\x02 \x01(\r:\x02\x38\x01\"K\n\x10PodStatsSequence\x12\x37\n\rpod_stats_map\x18\x01 \x03(\x0b\x32 .tensorflow.profiler.PodStatsMap\"\xf0\x01\n\x0b\x43hannelInfo\x12\x12\n\nchannel_id\x18\x01 \x01(\x03\x12\x14\n\x0csrc_core_ids\x18\x0b \x03(\r\x12\x14\n\x0c\x64st_core_ids\x18\x0c \x03(\r\x12\x11\n\tdata_size\x18\x04 \x01(\x04\x12\x13\n\x0b\x64uration_us\x18\x05 \x01(\x01\x12\x13\n\x0boccurrences\x18\x06 \x01(\r\x12\x13\n\x0butilization\x18\x07 \x01(\x01\x12\x11\n\thlo_names\x18\x08 \x03(\t\x12\x15\n\rsend_delay_us\x18\t \x01(\x01\x12\x13\n\x0b\x64\x65scription\x18\r \x01(\tJ\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04J\x04\x08\n\x10\x0b\"$\n\x10PodViewerSummary\x12\x10\n\x08warnings\x18\x01 \x03(\t\"\xe9\x01\n\x11PodViewerTopology\x12\x13\n\x0bx_dimension\x18\x01 \x01(\x05\x12\x13\n\x0by_dimension\x18\x02 \x01(\x05\x12\x13\n\x0bz_dimension\x18\x03 \x01(\x05\x12\x15\n\rhost_x_stride\x18\x04 \x01(\x05\x12\x15\n\rhost_y_stride\x18\x05 \x01(\x05\x12\x15\n\rhost_z_stride\x18\x06 \x01(\x05\x12\x1a\n\x12num_cores_per_chip\x18\x07 \x01(\x05\x12\x34\n\x05\x63ores\x18\x08 \x03(\x0b\x32%.tensorflow.profiler.TopologyLocation\"\xfb\x02\n\x11PodViewerDatabase\x12\x13\n\x0b\x64\x65vice_type\x18\n \x01(\t\x12\x41\n\x12pod_stats_sequence\x18\x03 \x01(\x0b\x32%.tensorflow.profiler.PodStatsSequence\x12\x36\n\x07summary\x18\x07 \x01(\x0b\x32%.tensorflow.profiler.PodViewerSummary\x12\x35\n\x0b\x64iagnostics\x18\x08 \x01(\x0b\x32 .tensorflow.profiler.Diagnostics\x12G\n\x15step_breakdown_events\x18\t \x03(\x0b\x32(.tensorflow.profiler.StepBreakdownEvents\x12\x38\n\x08topology\x18\x0b \x01(\x0b\x32&.tensorflow.profiler.PodViewerTopologyJ\x04\x08\x01\x10\x02J\x04\x08\x02\x10\x03J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.pod_viewer_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_PODSTATSMAP_PODSTATSPERCOREENTRY']._loaded_options = None
+  _globals['_PODSTATSMAP_PODSTATSPERCOREENTRY']._serialized_options = b'8\001'
+  _globals['_PODSTATSMAP_COREIDTOREPLICAIDMAPENTRY']._loaded_options = None
+  _globals['_PODSTATSMAP_COREIDTOREPLICAIDMAPENTRY']._serialized_options = b'8\001'
+  _globals['_REPLICAGROUP']._serialized_start=181
+  _globals['_REPLICAGROUP']._serialized_end=216
+  _globals['_ALLREDUCEOPINFO']._serialized_start=219
+  _globals['_ALLREDUCEOPINFO']._serialized_end=391
+  _globals['_PODSTATSMAP']._serialized_start=394
+  _globals['_PODSTATSMAP']._serialized_end=875
+  _globals['_PODSTATSMAP_PODSTATSPERCOREENTRY']._serialized_start=723
+  _globals['_PODSTATSMAP_PODSTATSPERCOREENTRY']._serialized_end=814
+  _globals['_PODSTATSMAP_COREIDTOREPLICAIDMAPENTRY']._serialized_start=816
+  _globals['_PODSTATSMAP_COREIDTOREPLICAIDMAPENTRY']._serialized_end=875
+  _globals['_PODSTATSSEQUENCE']._serialized_start=877
+  _globals['_PODSTATSSEQUENCE']._serialized_end=952
+  _globals['_CHANNELINFO']._serialized_start=955
+  _globals['_CHANNELINFO']._serialized_end=1195
+  _globals['_PODVIEWERSUMMARY']._serialized_start=1197
+  _globals['_PODVIEWERSUMMARY']._serialized_end=1233
+  _globals['_PODVIEWERTOPOLOGY']._serialized_start=1236
+  _globals['_PODVIEWERTOPOLOGY']._serialized_end=1469
+  _globals['_PODVIEWERDATABASE']._serialized_start=1472
+  _globals['_PODVIEWERDATABASE']._serialized_end=1851
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/power_metrics_pb2.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/power_metrics.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/power_metrics.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)plugin/xprof/protobuf/power_metrics.proto\x12\x13tensorflow.profiler\"\x8b\x02\n\x15PowerComponentMetrics\x12\x16\n\x0e\x63omponent_name\x18\x01 \x01(\t\x12\x11\n\tmax_power\x18\x02 \x01(\x01\x12\x11\n\tavg_power\x18\x03 \x01(\x01\x12\"\n\x1amax_moving_avg_power_100us\x18\x04 \x01(\x01\x12 \n\x18max_moving_avg_power_1ms\x18\x05 \x01(\x01\x12!\n\x19max_moving_avg_power_10ms\x18\x06 \x01(\x01\x12\x14\n\x0ctimescale_us\x18\x07 \x01(\r\x12\x14\n\x0csample_count\x18\x08 \x01(\x04\x12\x1f\n\x17max_moving_avg_power_1s\x18\t \x01(\x01\"[\n\x0cPowerMetrics\x12K\n\x17power_component_metrics\x18\x01 \x03(\x0b\x32*.tensorflow.profiler.PowerComponentMetricsb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.power_metrics_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_POWERCOMPONENTMETRICS']._serialized_start=67
+  _globals['_POWERCOMPONENTMETRICS']._serialized_end=334
+  _globals['_POWERMETRICS']._serialized_start=336
+  _globals['_POWERMETRICS']._serialized_end=427
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/roofline_model_pb2.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/roofline_model.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/roofline_model.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from xprof.protobuf import diagnostics_pb2 as plugin_dot_xprof_dot_protobuf_dot_diagnostics__pb2
+from xprof.protobuf import source_info_pb2 as plugin_dot_xprof_dot_protobuf_dot_source__info__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*plugin/xprof/protobuf/roofline_model.proto\x12\"tensorflow.profiler.roofline_model\x1a\'plugin/xprof/protobuf/diagnostics.proto\x1a\'plugin/xprof/protobuf/source_info.proto\"\xc4\x03\n\x15RooflineModelDatabase\x12\x13\n\x0b\x64\x65vice_type\x18\x01 \x01(\t\x12\x10\n\x08megacore\x18\x0c \x01(\x08\x12\x10\n\x08has_cmem\x18\x08 \x01(\x08\x12\x17\n\x0fhas_merged_vmem\x18\x0f \x01(\x08\x12\x16\n\x0epeak_flop_rate\x18\x02 \x01(\x01\x12\x13\n\x0bpeak_hbm_bw\x18\t \x01(\x01\x12\x19\n\x11peak_cmem_read_bw\x18\n \x01(\x01\x12\x1a\n\x12peak_cmem_write_bw\x18\x0b \x01(\x01\x12\x19\n\x11peak_vmem_read_bw\x18\r \x01(\x01\x12\x1a\n\x12peak_vmem_write_bw\x18\x0e \x01(\x01\x12V\n\x15roofline_model_record\x18\x05 \x03(\x0b\x32\x37.tensorflow.profiler.roofline_model.RooflineModelRecord\x12\x35\n\x0b\x64iagnostics\x18\x07 \x01(\x0b\x32 .tensorflow.profiler.Diagnostics\x12\x1d\n\x15time_scale_multiplier\x18\x10 \x01(\x01J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x06\x10\x07\"\x8d\n\n\x13RooflineModelRecord\x12\x43\n\x0brecord_type\x18\x12 \x01(\x0e\x32..tensorflow.profiler.roofline_model.RecordType\x12\x10\n\x08step_num\x18\x13 \x01(\r\x12\x0c\n\x04rank\x18\x01 \x01(\x04\x12\x15\n\rhlo_module_id\x18# \x01(\x04\x12\x14\n\x0chlo_category\x18\x11 \x01(\t\x12\x10\n\x08hlo_name\x18\x02 \x01(\t\x12\x13\n\x0boccurrences\x18\x03 \x01(\x03\x12\x18\n\x10total_time_in_us\x18\x04 \x01(\x01\x12!\n\x19total_time_per_core_in_us\x18\x14 \x01(\x01\x12 \n\x18total_time_in_percentage\x18\x15 \x01(\x01\x12\x16\n\x0e\x61vg_time_in_us\x18\x05 \x01(\x01\x12\x1d\n\x15total_self_time_in_us\x18\x06 \x01(\x01\x12\x1b\n\x13\x61vg_self_time_in_us\x18\x07 \x01(\x01\x12#\n\x1btotal_self_time_as_fraction\x18\x08 \x01(\x01\x12.\n&cumulative_total_self_time_as_fraction\x18\t \x01(\x01\x12\x1a\n\x12\x64ma_stall_fraction\x18\n \x01(\x01\x12\x1a\n\x12measured_flop_rate\x18\r \x01(\x01\x12\x17\n\x0fmodel_flop_rate\x18& \x01(\x01\x12\x1a\n\x12measured_memory_bw\x18\x0e \x01(\x01\x12\x0e\n\x06hbm_bw\x18\x1b \x01(\x01\x12\x14\n\x0c\x63mem_read_bw\x18\x1c \x01(\x01\x12\x15\n\rcmem_write_bw\x18\x1d \x01(\x01\x12\x14\n\x0cvmem_read_bw\x18\' \x01(\x01\x12\x15\n\rvmem_write_bw\x18( \x01(\x01\x12\x1d\n\x15operational_intensity\x18\x0f \x01(\x01\x12!\n\x19hbm_operational_intensity\x18\x1e \x01(\x01\x12\'\n\x1f\x63mem_read_operational_intensity\x18\x1f \x01(\x01\x12(\n cmem_write_operational_intensity\x18  \x01(\x01\x12\'\n\x1fvmem_read_operational_intensity\x18) \x01(\x01\x12(\n vmem_write_operational_intensity\x18* \x01(\x01\x12(\n bottleneck_operational_intensity\x18! \x01(\x01\x12\x10\n\x08\x62ound_by\x18\x10 \x01(\t\x12\x19\n\x11optimal_flop_rate\x18\x16 \x01(\x01\x12\x1b\n\x13roofline_efficiency\x18\" \x01(\x01\x12&\n\x1e\x66lop_rate_relative_to_hw_limit\x18\x18 \x01(\x01\x12&\n\x1ememory_bw_relative_to_hw_limit\x18\x19 \x01(\x01\x12\x1e\n\x16include_infeed_outfeed\x18\x1a \x01(\x08\x12#\n\x1b\x61pply_time_scale_multiplier\x18, \x01(\x08\x12\r\n\x05\x66lops\x18$ \x01(\x04\x12\x16\n\x0e\x62ytes_accessed\x18% \x01(\x04\x12\x34\n\x0bsource_info\x18+ \x01(\x0b\x32\x1f.tensorflow.profiler.SourceInfoJ\x04\x08\x0b\x10\x0cJ\x04\x08\x0c\x10\rJ\x04\x08\x17\x10\x18*Z\n\nRecordType\x12\x17\n\x13INVALID_RECORD_TYPE\x10\x00\x12\x07\n\x03\x41LL\x10\x01\x12\x10\n\x0c\x41VERAGE_STEP\x10\x02\x12\x0c\n\x08PER_STEP\x10\x03\x12\n\n\x06\x41LL_HW\x10\x04')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.roofline_model_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_RECORDTYPE']._serialized_start=1915
+  _globals['_RECORDTYPE']._serialized_end=2005
+  _globals['_ROOFLINEMODELDATABASE']._serialized_start=165
+  _globals['_ROOFLINEMODELDATABASE']._serialized_end=617
+  _globals['_ROOFLINEMODELRECORD']._serialized_start=620
+  _globals['_ROOFLINEMODELRECORD']._serialized_end=1913
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/smart_suggestion_pb2.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/smart_suggestion.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/smart_suggestion.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,plugin/xprof/protobuf/smart_suggestion.proto\x12\x13tensorflow.profiler\"=\n\x0fSmartSuggestion\x12\x11\n\trule_name\x18\x01 \x01(\t\x12\x17\n\x0fsuggestion_text\x18\x02 \x01(\t\"R\n\x15SmartSuggestionReport\x12\x39\n\x0bsuggestions\x18\x01 \x03(\x0b\x32$.tensorflow.profiler.SmartSuggestionb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.smart_suggestion_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_SMARTSUGGESTION']._serialized_start=69
+  _globals['_SMARTSUGGESTION']._serialized_end=130
+  _globals['_SMARTSUGGESTIONREPORT']._serialized_start=132
+  _globals['_SMARTSUGGESTIONREPORT']._serialized_end=214
+# @@protoc_insertion_point(module_scope)

xprof/protobuf/source_info_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: plugin/xprof/protobuf/source_info.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'plugin/xprof/protobuf/source_info.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'plugin/xprof/protobuf/source_info.proto\x12\x13tensorflow.profiler\"I\n\nSourceInfo\x12\x11\n\tfile_name\x18\x01 \x01(\t\x12\x13\n\x0bline_number\x18\x02 \x01(\x05\x12\x13\n\x0bstack_frame\x18\x03 \x01(\tb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'plugin.xprof.protobuf.source_info_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_SOURCEINFO']._serialized_start=64
+  _globals['_SOURCEINFO']._serialized_end=137
+# @@protoc_insertion_point(module_scope)