From 498483697afbe0b1875a416445db7f7c6f4f4a26 Mon Sep 17 00:00:00 2001 From: Yin Zhang Date: Fri, 13 Feb 2026 16:03:41 -0800 Subject: [PATCH] Internal only changes PiperOrigin-RevId: 869925450 --- plugin/trace_viewer/tf_trace_viewer/tf-trace-viewer.html | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/plugin/trace_viewer/tf_trace_viewer/tf-trace-viewer.html b/plugin/trace_viewer/tf_trace_viewer/tf-trace-viewer.html index b8d105e7..dec96ac0 100644 --- a/plugin/trace_viewer/tf_trace_viewer/tf-trace-viewer.html +++ b/plugin/trace_viewer/tf_trace_viewer/tf-trace-viewer.html @@ -813,9 +813,9 @@ `; }, - createPercaleLink: function(hloOp, host) { + createPercaleLink: function(hloOp, hloModule, host) { if (!this._percaleUrlPrefix || !this._sessionId || !hloOp) return; - const percaleLink = new URL(`${this._percaleUrlPrefix}/static?id=${this._sessionId}&hlo=${hloOp}`); + const percaleLink = new URL(`${this._percaleUrlPrefix}/static?id=${this._sessionId}&hlo=${hloOp}&hlo_module=${hloModule}`); if (host) { percaleLink.searchParams.append('host', host); } @@ -892,7 +892,7 @@ this.createCrossToolLink("roofline_model", "Roofline Model", {"roofline_op_name": hloOp}, rooflineText); const hostnamePrefix = currentProcess.name.split(' ')[0]; const host = this._selectedHosts.find(host => host.startsWith(hostnamePrefix)); - this.createPercaleLink(hloOp, host); + this.createPercaleLink(hloOp, hloModule, host); } else { // Add crosslink form trace viewer to graph viewer // per event selection for tensorboard profiler