feat: enhance capture window env

MooreThreads · Sep 19, 2024 · e8296fa · e8296fa
1 parent 4bb26ab
commit e8296fa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/libkineto/src/MuptiActivityProfiler.cpp b/libkineto/src/MuptiActivityProfiler.cpp
@@ -929,7 +929,10 @@ void MuptiActivityProfiler::stopTraceInternal(
         LOG(ERROR) << "Invalid value for MT_TIMER_CAPTURE_WINDOW_LEN. Using default 60s.\n";
       }
     }
-    captureWindowEndTime_ = captureWindowStartTime_ + 1000000*captureWindowLen;
+    if (captureWindowLen > 0) {
+      captureWindowEndTime_ = captureWindowStartTime_ + 1000000*captureWindowLen;
+    }
+    LOG(INFO) << "MT_TIMER_GPU_EVENTS enabled, capture window length: " << captureWindowEndTime_ - captureWindowStartTime_ << "us";
   }
 #if defined(HAS_MUPTI) || defined(HAS_ROCTRACER)
   if (!cpuOnly_) {