Skip to content

Commit 91ec61a

Browse files
committed
Fix up naming
1 parent 6ab2ba7 commit 91ec61a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/libtorch.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ namespace {
6565

6666
#ifdef TRITON_ENABLE_GPU
6767
void CUDART_CB
68-
CaptureLastTimestampCallback(void* data)
68+
CaptureTimestampCallback(void* data)
6969
{
7070
auto* timestamp = reinterpret_cast<std::atomic<uint64_t>*>(data);
7171
SET_TIMESTAMP(*timestamp);
@@ -1312,7 +1312,7 @@ ModelInstanceState::ProcessRequests(
13121312
// For the compute infer duration, multiple streams will be involved, so we
13131313
// need to launch a CUDA callback function for timestamp capturing.
13141314
cudaLaunchHostFunc(
1315-
GetCudaStreamByInstanceKind(), CaptureLastTimestampCallback,
1315+
GetCudaStreamByInstanceKind(), CaptureTimestampCallback,
13161316
reinterpret_cast<void*>(&compute_output_start));
13171317
#endif
13181318
} else {

0 commit comments

Comments
 (0)