Update links to graphs

PiperOrigin-RevId: 720749132
tensorflow · Jan 29, 2025 · 30cd3eb · 30cd3eb
1 parent 3fe152b
commit 30cd3eb
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/frontend/app/components/op_profile/op_details/op_details.ng.html b/frontend/app/components/op_profile/op_details/op_details.ng.html
@@ -25,6 +25,20 @@
             </button>
           </a>
         </div>
+        <div class="info" id="graph-viewer-link" [hidden]="isOss || !isCustomCall">
+          <a [href]="getCustomCallTextLink()" target="_blank" style="text-decoration:none;">
+            <button
+              mat-stroked-button
+              extended
+              type="button"
+              color="primary"
+              appearance="fill"
+              aria-label="go to custom call text"
+              i18n-aria-label="A button to corresponding custom call text page">
+             > Custom Call Text
+            </button>
+          </a>
+        </div>
         <div class="info" [hidden]="!avgTimeMs">
           <div class="title">Total Time Avg:</div>
           <code class="expression">{{avgTimeMs}}</code>

diff --git a/frontend/app/components/op_profile/op_details/op_details.ts b/frontend/app/components/op_profile/op_details/op_details.ts
@@ -64,6 +64,9 @@ export class OpDetails {
   showUtilizationWarning = false;
   deviceType = 'TPU';
 
+  get isCustomCall() {
+    return this.node?.xla?.category === 'custom-call';
+  }
 
   constructor(
       private readonly store: Store<{}>,
@@ -150,6 +153,12 @@ export class OpDetails {
     return '';
   }
 
+  getCustomCallTextLink() {
+    return `/graph_viewer.json?session_id=${this.sessionId}&module_name=${
+        this.getSelectedModuleName()}&node_name=${
+        this.getSelectedOpName()}&type=custom_call`;
+  }
+
   dimensionColor(dimension?: Node.XLAInstruction.LayoutAnalysis.Dimension):
       string {
     if (!dimension || !dimension.alignment) {