Skip to content

Commit

Permalink
Store memory_viewer_preprocess protobuf ts file as plain text instead…
Browse files Browse the repository at this point in the history
… of gz repo.

This makes it easier to do updates and code reviews.

PiperOrigin-RevId: 687082345
  • Loading branch information
Profiler Team authored and copybara-github committed Oct 17, 2024
1 parent 12deb53 commit 5ac49e3
Show file tree
Hide file tree
Showing 3 changed files with 63 additions and 10 deletions.
10 changes: 0 additions & 10 deletions frontend/app/common/interfaces/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -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 = [
Expand Down
Original file line number Diff line number Diff line change
@@ -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;
}
Binary file not shown.

0 comments on commit 5ac49e3

Please sign in to comment.