Skip to content

Commit fdc2747

Browse files
[XLA:GPU]: Add peer to peer copies for cupti tracing.
Before this change peer to peer copies done using cuMemcpyPeerAsync was not being tracked with the driver API. PiperOrigin-RevId: 825040246
1 parent 40820a5 commit fdc2747

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

xla/backends/profiler/gpu/cupti_tracer.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1225,6 +1225,8 @@ CuptiTracer::CreateDefaultCallbackIds() {
12251225
CUPTI_DRIVER_TRACE_CBID_cuMemcpy3DAsync_v2,
12261226
CUPTI_DRIVER_TRACE_CBID_cuMemcpyHtoA_v2,
12271227
CUPTI_DRIVER_TRACE_CBID_cuMemcpyHtoAAsync_v2,
1228+
CUPTI_DRIVER_TRACE_CBID_cuMemcpyPeer,
1229+
CUPTI_DRIVER_TRACE_CBID_cuMemcpyPeerAsync,
12281230
// MemAlloc
12291231
CUPTI_DRIVER_TRACE_CBID_cuMemAlloc_v2,
12301232
CUPTI_DRIVER_TRACE_CBID_cuMemAllocPitch_v2,

0 commit comments

Comments
 (0)