Fix P90 math for hlo op profile

PiperOrigin-RevId: 569350204
tensorflow · Sep 29, 2023 · e79eb6a · e79eb6a
1 parent 1219c5d
commit e79eb6a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frontend/app/components/op_profile/op_table_entry/op_table_entry.ts b/frontend/app/components/op_profile/op_table_entry/op_table_entry.ts
@@ -109,10 +109,10 @@ export class OpTableEntry implements OnChanges {
     }
 
     let tot = 0;
-    const target90 = utils.timeFraction(this.node, this.rootNode) * 0.9;
+    const targetP90NodeRawTimePs = this.node.metrics.rawTime * 0.9;
     const targetCount = Math.min(this.childrenCount, this.node.children.length);
     for (let i = 0; i < targetCount; i++) {
-      if (tot >= target90) {
+      if (tot >= targetP90NodeRawTimePs) {
         return i;
       }
       const child = this.node.children[i];