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