Skip to content

Commit 4c29311

Browse files
committed
Fix parcagpu.bt script
1 parent a86187f commit 4c29311

1 file changed

Lines changed: 13 additions & 15 deletions

File tree

parcagpu.bt

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -11,24 +11,22 @@ BEGIN
1111
printf("Monitoring parcagpu USDT probes...\n\n");
1212
}
1313

14-
// Track graph launches
15-
// Arguments: start, end, correlationId|deviceId, streamId, graphId
16-
usdt:cupti/build/libparcagpucupti.so:parcagpu:graph_executed
14+
15+
16+
// Track CUDA API correlation events
17+
// Arguments: correlationId (uint32), signedCbid (int32), name (char*)
18+
usdt:*:parcagpu:cuda_correlation
1719
{
18-
$start = arg0;
19-
$end = arg1;
20-
printf("[%d] Graph executed:\n", pid);
21-
printf(" start=%lu, end=%lu, duration=%lu ns\n", $start, $end, $end - $start);
22-
$devId = arg2 >> 32;
23-
$corId = (uint64)arg2 & 0xFFFFFFFF;
24-
printf(" correlationId=%u, deviceId=%u, streamId=%u, graphId=%u\n",
25-
$corId, $devId, arg3, arg4);
26-
@graph_launches = count();
20+
$corId = (uint32)arg0;
21+
$cbid = (int32)arg1;
22+
printf("[%d] cuda_correlation: correlationId=%u, cbid=%d, name=%s\n",
23+
pid, $corId, $cbid, str(arg2));
24+
@cuda_correlations = count();
2725
}
2826

2927
// Track kernel executions
3028
// Arguments: start, end, correlationId, deviceId, streamId, name (pointer)
31-
usdt:cupti/build/libparcagpucupti.so:parcagpu:kernel_executed
29+
usdt:*:parcagpu:kernel_executed
3230
{
3331
$start = arg0;
3432
$end = arg1;
@@ -45,8 +43,8 @@ usdt:cupti/build/libparcagpucupti.so:parcagpu:kernel_executed
4543
END
4644
{
4745
printf("\n=== Summary ===\n");
48-
printf("Graph executions: ");
49-
print(@graph_launches);
46+
printf("CUDA correlations: ");
47+
print(@cuda_correlations);
5048
printf("Kernel executions: ");
5149
print(@kernel_executions);
5250
}

0 commit comments

Comments
 (0)