diff --git a/nvtx_pmpi_wrappers/nvtx.w b/nvtx_pmpi_wrappers/nvtx.w index e3282b1..8e366a6 100644 --- a/nvtx_pmpi_wrappers/nvtx.w +++ b/nvtx_pmpi_wrappers/nvtx.w @@ -27,7 +27,7 @@ #include nvtxDomainHandle_t nvtx_mpi_domain; // Setup event category name -{{fn name MPI_Init}} +{{fn name MPI_Init MPI_Init_thread}} nvtx_mpi_domain = nvtxDomainCreateA("MPI"); {{callfn}} {{endfn}}