From 5ac49e3d50ac04c0af89ff2bb8145d42c5838d18 Mon Sep 17 00:00:00 2001 From: Profiler Team Date: Thu, 17 Oct 2024 16:25:25 -0700 Subject: [PATCH] Store memory_viewer_preprocess protobuf ts file as plain text instead of gz repo. This makes it easier to do updates and code reviews. PiperOrigin-RevId: 687082345 --- frontend/app/common/interfaces/BUILD | 10 --- ...memory_viewer_preprocess.jsonpb_decls.d.ts | 63 ++++++++++++++++++ ...ory_viewer_preprocess.jsonpb_decls.d.ts.gz | Bin 517 -> 0 bytes 3 files changed, 63 insertions(+), 10 deletions(-) create mode 100644 frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts delete mode 100644 frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts.gz diff --git a/frontend/app/common/interfaces/BUILD b/frontend/app/common/interfaces/BUILD index 426319362..d748e5bd7 100644 --- a/frontend/app/common/interfaces/BUILD +++ b/frontend/app/common/interfaces/BUILD @@ -32,16 +32,6 @@ genrule( ], ) -genrule( - name = "memory_viewer_preprocess_proto_ts", - srcs = ["memory_viewer_preprocess.jsonpb_decls.d.ts.gz"], - outs = ["memory_viewer_preprocess.jsonpb_decls.d.ts"], - cmd = "gzip -d -c $(SRCS) > $(OUTS)", - visibility = [ - "//frontend:__subpackages__", "//plugin:__subpackages__", - ], -) - ts_library( name = "op_metrics_proto_defs", srcs = [ diff --git a/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts b/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts new file mode 100644 index 000000000..75d5cb670 --- /dev/null +++ b/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts @@ -0,0 +1,63 @@ +// This file must be kept in sync with the corresponding proto file. + +/** See corresponding proto */ +export interface HeapObject { + numbered?: /* int32 */ number; + named?: string; + label?: string; + logicalBufferId?: /* int32 */ number; + logicalBufferSizeMib?: /* double */ number; + unpaddedShapeMib?: /* double */ number; + instructionName?: string; + shapeString?: string; + tfOpName?: string; + groupName?: string; + opCode?: string; +} + +/** See corresponding proto */ +export interface BufferSpan { + start?: /* int32 */ number; + limit?: /* int32 */ number; +} + +/** See corresponding proto */ +export interface LogicalBuffer { + id?: /* int64 */ string; + shape?: string; + sizeMib?: /* double */ number; + hloName?: string; + shapeIndex?: /* int64 */ string[]; +} + +/** See corresponding proto */ +export interface BufferAllocation { + id?: /* int64 */ string; + sizeMib?: /* double */ number; + attributes?: string[]; + logicalBuffers?: LogicalBuffer[]; + commonShape?: string; +} + +/** See corresponding proto */ +export interface PreprocessResult { + heapSizes?: /* double */ number[]; + unpaddedHeapSizes?: /* double */ number[]; + maxHeap?: HeapObject[]; + maxHeapBySize?: HeapObject[]; + logicalBufferSpans?: {[key: /* int32 */ string]: BufferSpan}; + maxHeapToBySize?: /* int32 */ number[]; + bySizeToMaxHeap?: /* int32 */ number[]; + moduleName?: string; + entryComputationName?: string; + peakHeapMib?: /* double */ number; + peakUnpaddedHeapMib?: /* double */ number; + peakHeapSizePosition?: /* int32 */ number; + entryComputationParametersMib?: /* double */ number; + nonReusableMib?: /* double */ number; + maybeLiveOutMib?: /* double */ number; + totalBufferAllocationMib?: /* double */ number; + indefiniteBufferAllocationMib?: /* double */ number; + indefiniteLifetimes?: BufferAllocation[]; + allocationTimeline?: string; +} diff --git a/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts.gz b/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts.gz deleted file mode 100644 index 610feddfa68bbd24a43498b3a8866f8c37198dd4..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 517 zcmV+g0{Zo5?!^DA=Zg7iQfc!ww8LZ$Lr(Q-l@Hu0u4*j~#Ys0H!gv74rC z+_bKUl*{hKGoGDIQQ$(8aRpZnO$`+m4+wfxzG3Bxud^aN%F7ZBvVS+l#d!o?{wdBc z*5$tq1mvo3o#ATM)C80$(i+Wp1>(-vHJZV(6FbNW|HMaLHgHD16qw?@(!dz9$sBYJ z%T>_iE5}Lw55g0THl8q96_SptN4n>OnIYd-Cw)(>Tl^01p5f~|YEWy#tpnqZdBY_? z-0$TW?%KJg;amT&ui_bdE~L)n$1oRk$kKsgxaehkdChopD+E;#=XH|wyi9NbUwVgj zqqGr`1TVHb*0rdLq$D-jGpY9yV}sgIh1Ncy^`hOYd04cVY}>E2rD~J?a2i~~A|eB5 zyU0oJZW+(%MReCLL^85{y}aYH8@J}huQxm4{qAr~X~VJ~)0VAVL8kP$(R>&yiFtuN z;X>ui@}8vj4i2oeMtF||nN&pB^N!mmz}umW$?`~kWNTBMV=zH!$cWAGC{j=8EreDE zDPdXSkbmIFyA0HkYsb9ZuIzu4Va*kH`0Edcyhg|6D#Ym_?WAE-nFgfbDqXQ3=2YGw HwFUqHY)1sO