@@ -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
4543END
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