diff --git a/libkineto/src/CuptiActivityProfiler.cpp b/libkineto/src/CuptiActivityProfiler.cpp index 286d5b359..e02c325c1 100644 --- a/libkineto/src/CuptiActivityProfiler.cpp +++ b/libkineto/src/CuptiActivityProfiler.cpp @@ -397,6 +397,9 @@ void CuptiActivityProfiler::processCpuTrace( const std::unique_ptr>::value, "handleActivity is unsafe and relies on the caller to maintain not " "only lifetime but also address stability."); + if (act->type() == ActivityType::USER_ANNOTATION && act->duration()<=0){ + act->endTime = captureWindowEndTime_; + } logger.handleActivity(*act); } clientActivityTraceMap_[act->correlationId()] = &span_pair;