Automated rollback of commit b88c97591c0399e111ab64c82e3f11a23aa9955c
PiperOrigin-RevId: 245156681
This commit is contained in:
parent
5dcd9cb4d6
commit
02bd711c79
@ -175,18 +175,15 @@ class CuptiCallbackHook {
|
||||
Status Enable(CudaEventRecorder* recorder) {
|
||||
TF_RETURN_IF_ERROR(
|
||||
ToStatus(cuptiSubscribe(&subscriber_, &CuptiCallback, recorder)));
|
||||
for (auto cbid :
|
||||
{CUPTI_DRIVER_TRACE_CBID_cuLaunchKernel,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuLaunchCooperativeKernelMultiDevice,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuLaunchCooperativeKernel,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpy,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyAsync,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyHtoD_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyHtoDAsync_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoH_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoHAsync_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoD_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoDAsync_v2}) {
|
||||
for (auto cbid : {CUPTI_DRIVER_TRACE_CBID_cuLaunchKernel,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpy,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyAsync,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyHtoD_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyHtoDAsync_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoH_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoHAsync_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoD_v2,
|
||||
CUPTI_DRIVER_TRACE_CBID_cuMemcpyDtoDAsync_v2}) {
|
||||
TF_RETURN_IF_ERROR(ToStatus(cuptiEnableCallback(
|
||||
/*enable=*/1, subscriber_, CUPTI_CB_DOMAIN_DRIVER_API, cbid)));
|
||||
}
|
||||
@ -244,11 +241,7 @@ class CuptiCallbackHook {
|
||||
const CUpti_CallbackData& cbdata,
|
||||
CudaEventRecorder* recorder) {
|
||||
switch (cbid) {
|
||||
case CUPTI_DRIVER_TRACE_CBID_cuLaunchKernel:
|
||||
TF_FALLTHROUGH_INTENDED;
|
||||
case CUPTI_DRIVER_TRACE_CBID_cuLaunchCooperativeKernelMultiDevice:
|
||||
TF_FALLTHROUGH_INTENDED;
|
||||
case CUPTI_DRIVER_TRACE_CBID_cuLaunchCooperativeKernel: {
|
||||
case CUPTI_DRIVER_TRACE_CBID_cuLaunchKernel: {
|
||||
DCHECK_NE(cbdata.symbolName, nullptr);
|
||||
auto params =
|
||||
static_cast<const cuLaunchKernel_params*>(cbdata.functionParams);
|
||||
|
Loading…
x
Reference in New Issue
Block a user