From e102e0dfa95e45b49c5301ce06dbab5234176ff3 Mon Sep 17 00:00:00 2001
From: Vivek Kale <11766050+vlkale@users.noreply.github.com>
Date: Thu, 20 Jul 2023 15:14:01 -0700
Subject: [PATCH] Update kp_nvtx_focused_connector.cpp

Fixing function names to be nvtx
---
 .../kp_nvtx_focused_connector.cpp                | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/profiling/nvprof-focused-connector/kp_nvtx_focused_connector.cpp b/profiling/nvprof-focused-connector/kp_nvtx_focused_connector.cpp
index dba824257..9bd1b40e1 100644
--- a/profiling/nvprof-focused-connector/kp_nvtx_focused_connector.cpp
+++ b/profiling/nvprof-focused-connector/kp_nvtx_focused_connector.cpp
@@ -23,15 +23,15 @@
 #include <cxxabi.h>
 #include <cuda_profiler_api.h>
 
-#include "kp_nvprof_focused_connector_domain.h"
+#include "kp_nvtx_focused_connector_domain.h"
 
 #include "kp_core.hpp"
 
 namespace KokkosTools {
-namespace NVProfFocusedConnector {
+namespace NVTXFocusedConnector {
 
-static KernelNVProfFocusedConnectorInfo* currentKernel;
-static std::unordered_map<std::string, KernelNVProfFocusedConnectorInfo*>
+static KernelNVTXFocusedConnectorInfo* currentKernel;
+static std::unordered_map<std::string, KernelNVTXFocusedConnectorInfo*>
     domain_map;
 static uint64_t nextKernelID;
 
@@ -40,7 +40,7 @@ void kokkosp_init_library(
     struct Kokkos_Profiling_KokkosPDeviceInfo* deviceInfo) {
   printf("-----------------------------------------------------------\n");
   printf(
-      "KokkosP: NVProf Analyzer Focused Connector (sequence is %d, version: "
+      "KokkosP: NVTX Analyzer Focused Connector (sequence is %d, version: "
       "%llu)\n",
       loadSeq, interfaceVer);
   printf("-----------------------------------------------------------\n");
@@ -48,14 +48,14 @@ void kokkosp_init_library(
   nextKernelID = 0;
 }
 
-KernelNVProfFocusedConnectorInfo* getFocusedConnectorInfo(
+KernelNVTXFocusedConnectorInfo* getFocusedConnectorInfo(
     const char* name, KernelExecutionType kType) {
   std::string nameStr(name);
   auto kDomain  = domain_map.find(nameStr);
   currentKernel = NULL;
 
   if (kDomain == domain_map.end()) {
-    currentKernel = new KernelNVProfFocusedConnectorInfo(name, kType);
+    currentKernel = new KernelNVTXFocusedConnectorInfo(name, kType);
     domain_map.insert(std::pair<std::string, KernelNVProfFocusedConnectorInfo*>(
         nameStr, currentKernel));
   } else {
@@ -138,7 +138,7 @@ Kokkos::Tools::Experimental::EventSet get_event_set() {
 
 extern "C" {
 
-namespace impl = KokkosTools::NVProfFocusedConnector;
+namespace impl = KokkosTools::NVTXFocusedConnector;
 
 EXPOSE_INIT(impl::kokkosp_init_library)
 EXPOSE_FINALIZE(impl::kokkosp_finalize_library)