File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -65,7 +65,7 @@ namespace {
65
65
66
66
#ifdef TRITON_ENABLE_GPU
67
67
void CUDART_CB
68
- CaptureLastTimestampCallback (void * data)
68
+ CaptureTimestampCallback (void * data)
69
69
{
70
70
auto * timestamp = reinterpret_cast <std::atomic<uint64_t >*>(data);
71
71
SET_TIMESTAMP (*timestamp);
@@ -1312,7 +1312,7 @@ ModelInstanceState::ProcessRequests(
1312
1312
// For the compute infer duration, multiple streams will be involved, so we
1313
1313
// need to launch a CUDA callback function for timestamp capturing.
1314
1314
cudaLaunchHostFunc (
1315
- GetCudaStreamByInstanceKind (), CaptureLastTimestampCallback ,
1315
+ GetCudaStreamByInstanceKind (), CaptureTimestampCallback ,
1316
1316
reinterpret_cast <void *>(&compute_output_start));
1317
1317
#endif
1318
1318
} else {
You can’t perform that action at this time.
0 commit comments