Skip to content

Commit

Permalink
[PROTON] Support kernels launched with cudaLaunchKernelExC and `cuL…
Browse files Browse the repository at this point in the history
…aunchKernelEx` on sm>=90 (#3668)

e.g., cublas/cutlass gemm kernels like
`sm90_xmma_gemm_f16f16_f16f32_f32_tn_n_tilesize128x128x64_warpgroupsize1x1x1_execute_segment_k_off_kernel__5x_cublas`
  • Loading branch information
Jokeren authored Apr 15, 2024
1 parent 3418789 commit 3657381
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions third_party/proton/csrc/lib/Profiler/CuptiProfiler.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -177,6 +177,8 @@ std::pair<bool, bool> matchKernelCbId(CUpti_CallbackId cbId) {
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunchKernel_v7000:
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunch_ptsz_v7000:
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunchKernel_ptsz_v7000:
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunchKernelExC_v11060:
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunchKernelExC_ptsz_v11060:
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunchCooperativeKernel_v9000:
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunchCooperativeKernel_ptsz_v9000:
case CUPTI_RUNTIME_TRACE_CBID_cudaLaunchCooperativeKernelMultiDevice_v9000: {
Expand All @@ -188,6 +190,8 @@ std::pair<bool, bool> matchKernelCbId(CUpti_CallbackId cbId) {
case CUPTI_DRIVER_TRACE_CBID_cuLaunchGridAsync:
case CUPTI_DRIVER_TRACE_CBID_cuLaunchKernel:
case CUPTI_DRIVER_TRACE_CBID_cuLaunchKernel_ptsz:
case CUPTI_DRIVER_TRACE_CBID_cuLaunchKernelEx:
case CUPTI_DRIVER_TRACE_CBID_cuLaunchKernelEx_ptsz:
case CUPTI_DRIVER_TRACE_CBID_cuLaunchCooperativeKernel:
case CUPTI_DRIVER_TRACE_CBID_cuLaunchCooperativeKernel_ptsz:
case CUPTI_DRIVER_TRACE_CBID_cuLaunchCooperativeKernelMultiDevice: {
Expand Down

0 comments on commit 3657381

Please sign in to comment.