From bf410d1747646bdce69cdab76264eba6ee3358ae Mon Sep 17 00:00:00 2001 From: Ammar ELWazir Date: Mon, 2 Aug 2021 16:59:27 -0400 Subject: [PATCH] SWDEV-294319 & SWDEV-294321 Added support for missing functions Added Support for launch kernel functions to fill_api_db Change-Id: Ie8749806214d6e283c3d408d8019f178f42018bf [ROCm/roctracer commit: 8396836c25daf521e433679efbfcf6de6617b68c] --- projects/roctracer/test/tool/tracer_tool.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/projects/roctracer/test/tool/tracer_tool.cpp b/projects/roctracer/test/tool/tracer_tool.cpp index b75fa6d8c1..8656dabf58 100644 --- a/projects/roctracer/test/tool/tracer_tool.cpp +++ b/projects/roctracer/test/tool/tracer_tool.cpp @@ -374,6 +374,7 @@ roctracer::TraceBuffer* 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;