Merge "SWDEV-294319 & SWDEV-294321 Added support for missing functions" into amd-staging

[ROCm/roctracer commit: 02d27d06e5]
This commit is contained in:
Ammar Elwazir
2021-08-12 03:15:40 -04:00
zatwierdzone przez Gerrit Code Review
@@ -374,6 +374,7 @@ roctracer::TraceBuffer<hip_api_trace_entry_t>* hip_api_trace_buffer = NULL;
static inline bool is_hip_kernel_launch_api(const uint32_t& cid) {
bool ret =
(cid == HIP_API_ID_hipLaunchKernel) ||
(cid == HIP_API_ID_hipExtLaunchKernel) ||
(cid == HIP_API_ID_hipLaunchCooperativeKernel) ||
(cid == HIP_API_ID_hipLaunchCooperativeKernelMultiDevice) ||
(cid == HIP_API_ID_hipExtLaunchMultiKernelMultiDevice) ||
@@ -439,6 +440,13 @@ void hip_api_callback(
if (f != NULL) entry->name = strdup(roctracer::HipLoader::Instance().KernelNameRefByPtr(f, stream));
break;
}
case HIP_API_ID_hipExtLaunchKernel:
{
const void* f = data->args.hipExtLaunchKernel.function_address;
hipStream_t stream = data->args.hipExtLaunchKernel.stream;
if (f != NULL) entry->name = strdup(roctracer::HipLoader::Instance().KernelNameRefByPtr(f, stream));
break;
}
default:
{
const hipFunction_t f = data->args.hipModuleLaunchKernel.f;