Fixed indexing

triton-inference-server · Jul 2, 2024 · b6ac93e · b6ac93e
1 parent 55369e2
commit b6ac93e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/identity.cc b/src/identity.cc
@@ -905,8 +905,8 @@ TRITONBACKEND_ModelInstanceExecute(
         }
       }
 
-      for (uint32_t count = 0; count < nesting_count; count++){
-        std::string end_span_str = std::string("CUSTOM_ACTIVITY" + std::to_string(count) + "_END");
+      for (uint32_t count = 0; count < nesting_count ; count++){
+        std::string end_span_str = std::string("CUSTOM_ACTIVITY" + std::to_string(nesting_count - count - 1 ) + "_END");
         SET_TIMESTAMP(custom_activity_ns);
         GUARDED_RESPOND_IF_ERROR(
           responses, r, TRITONSERVER_InferenceTraceReportActivity(trace, custom_activity_ns, end_span_str.c_str()));