Lines Matching refs:cbdata
197 const void* cbdata) {
199 auto data = static_cast<const CUpti_CallbackData*>(cbdata);
223 const CUpti_CallbackData& cbdata,
225 auto params = static_cast<const T*>(cbdata.functionParams);
226 *cbdata.correlationData = recorder->StartMemcpy(
227 src_type, dst_type, params->ByteCount, cbdata.context, nullptr);
231 const CUpti_CallbackData& cbdata,
233 auto params = static_cast<const T*>(cbdata.functionParams);
234 *cbdata.correlationData = recorder->StartMemcpy(
235 src_type, dst_type, params->ByteCount, cbdata.context, params->hStream);
239 const CUpti_CallbackData& cbdata,
243 DCHECK_NE(cbdata.symbolName, nullptr);
245 static_cast<const cuLaunchKernel_params*>(cbdata.functionParams);
246 *cbdata.correlationData = recorder->StartKernel(
247 cbdata.symbolName, cbdata.context, params->hStream);
253 static_cast<const cuMemcpy_params*>(cbdata.functionParams);
255 GetMemoryType(params->dst), cbdata,
260 static_cast<const cuMemcpyAsync_params*>(cbdata.functionParams);
262 GetMemoryType(params->src), GetMemoryType(params->dst), cbdata,
268 CU_MEMORYTYPE_HOST, CU_MEMORYTYPE_DEVICE, cbdata, recorder);
272 CU_MEMORYTYPE_HOST, CU_MEMORYTYPE_DEVICE, cbdata, recorder);
276 CU_MEMORYTYPE_DEVICE, CU_MEMORYTYPE_HOST, cbdata, recorder);
279 CU_MEMORYTYPE_DEVICE, CU_MEMORYTYPE_HOST, cbdata, recorder);
283 CU_MEMORYTYPE_DEVICE, CU_MEMORYTYPE_DEVICE, cbdata, recorder);
286 CU_MEMORYTYPE_DEVICE, CU_MEMORYTYPE_DEVICE, cbdata, recorder);
294 const CUpti_CallbackData& cbdata,
298 recorder->StopKernel(*cbdata.correlationData);
308 recorder->StopMemcpy(*cbdata.correlationData);