add profile start time to tensorboard profiler overview

PiperOrigin-RevId: 622975506
tensorflow · Apr 8, 2024 · f7bf1c8 · f7bf1c8
1 parent 30fc866
commit f7bf1c8
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/frontend/app/common/interfaces/data_table.ts b/frontend/app/common/interfaces/data_table.ts
@@ -85,13 +85,14 @@ export declare interface TopOpsColumn {
 
 /* tslint:disable enforce-name-casing */
 /** The base interface for a property of run environment. */
-declare interface RunEnvironmentProperty {
+export declare interface RunEnvironmentProperty {
   error_message?: string;
   host_count?: string;
   device_core_count?: string;
   device_type?: string;
   is_training?: string;
   tpu_core_count?: string;
+  profile_start_time?: string;
 }
 /* tslint:enable */
 

diff --git a/frontend/app/components/overview/overview.ng.html b/frontend/app/components/overview/overview.ng.html
@@ -18,7 +18,14 @@
           [generalAnalysis]="generalAnalysis"
           [inputPipelineAnalysis]="inputPipelineAnalysis">
       </performance-summary>
-      <run-environment-view [runEnvironment]="runEnvironment"></run-environment-view>
+      <run-environment-view [runEnvironment]="runEnvironment">
+        <div profileDetail>
+          <p [hidden]="!profileStartTime">
+            <span class="key-label">Profile start time (local)</span>:
+            <span>{{profileStartTime}}</span>
+          </p>
+        </div>
+      </run-environment-view>
     </div>
 
     <div class="right-column">

diff --git a/frontend/app/components/overview/overview.ts b/frontend/app/components/overview/overview.ts
@@ -1,7 +1,7 @@
 import {Component, OnDestroy} from '@angular/core';
 import {ActivatedRoute} from '@angular/router';
 import {Store} from '@ngrx/store';
-import {OverviewDataTuple} from 'org_xprof/frontend/app/common/interfaces/data_table';
+import {OverviewDataTuple, RunEnvironmentProperty} from 'org_xprof/frontend/app/common/interfaces/data_table';
 import {NavigationEvent} from 'org_xprof/frontend/app/common/interfaces/navigation_event';
 import {DataService} from 'org_xprof/frontend/app/services/data_service/data_service';
 import {setLoadingStateAction} from 'org_xprof/frontend/app/store/actions';
@@ -20,6 +20,8 @@ export class Overview extends OverviewCommon implements OnDestroy {
   /** Handles on-destroy Subject, used to unsubscribe. */
   private readonly destroyed = new ReplaySubject<void>(1);
 
+  profileStartTime = '';
+
   constructor(
       route: ActivatedRoute, private readonly dataService: DataService,
       private readonly store: Store<{}>) {
@@ -53,9 +55,16 @@ export class Overview extends OverviewCommon implements OnDestroy {
 
           /** Transfer data to Overview DataTable type */
           this.parseOverviewData((data || []) as OverviewDataTuple);
+          this.parseRunEnvironmentDetail();
         });
   }
 
+  parseRunEnvironmentDetail() {
+    const runEnvironmentProp: RunEnvironmentProperty =
+        (this.runEnvironment || {}).p || {};
+    this.profileStartTime = runEnvironmentProp.profile_start_time || '';
+  }
+
   ngOnDestroy() {
     // Unsubscribes all pending subscriptions.
     this.destroyed.next();