diff --git a/tensorflow/core/platform/default/device_tracer.cc b/tensorflow/core/platform/default/device_tracer.cc index ee41bb576c2..44dd631dbbc 100644 --- a/tensorflow/core/platform/default/device_tracer.cc +++ b/tensorflow/core/platform/default/device_tracer.cc @@ -231,7 +231,7 @@ class CuptiCallbackHook { src_type, dst_type, params->ByteCount, cbdata.context, nullptr); } template - static void StartMemcpyAsync(CUmemorytype dst_type, CUmemorytype src_type, + static void StartMemcpyAsync(CUmemorytype src_type, CUmemorytype dst_type, const CUpti_CallbackData& cbdata, CudaEventRecorder* recorder) { auto params = static_cast(cbdata.functionParams);