From 06d918d2b4069456d2393e71eb32c5b89f190cdd Mon Sep 17 00:00:00 2001
From: gobhardw <gopesh.bhardwaj@amd.com>
Date: Wed, 26 Jul 2023 00:18:31 +0530
Subject: [PATCH] Adding kokkos kernel name trucate test

Change-Id: I01af7dbb399cfe470e548a584faec890924d7039
---
 tests-v2/unittests/profiler_gtest.cpp | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/tests-v2/unittests/profiler_gtest.cpp b/tests-v2/unittests/profiler_gtest.cpp
index b7c520bb8a..9c62ccd4d3 100644
--- a/tests-v2/unittests/profiler_gtest.cpp
+++ b/tests-v2/unittests/profiler_gtest.cpp
@@ -331,4 +331,25 @@ TEST(WhenTrucatingLongKernelNames, KernelNameGetsTruncatedProperly) {
   std::string trunkated_name = rocprofiler::truncate_name(long_kernel_name);
 
   EXPECT_EQ("kernel_7r_3d_pml", trunkated_name);
+}
+
+TEST(WhenTrucatingKokkossKernelNames, KernelNameGetsTruncatedProperly) {
+  std::string long_kernel_name =
+      "void "
+      "Kokkos::Experimental::Impl::hip_parallel_launch_local_memory<Kokkos::Impl::ParallelFor<"
+      "Kokkos::"
+      "Impl::ViewValueFunctor<Kokkos::Device<Kokkos::Experimental::HIP, "
+      "Kokkos::Experimental::HIPSpace>, "
+      "LAMMPS_NS::PairReaxFFKokkos<Kokkos::Experimental::HIP>::params_sing, false>, "
+      "Kokkos::RangePolicy<Kokkos::Experimental::HIP, Kokkos::IndexType<long> >, "
+      "Kokkos::Experimental::HIP>, 1024u, "
+      "1u>(Kokkos::Impl::ParallelFor<Kokkos::Impl::ViewValueFunctor<Kokkos::Device<Kokkos::"
+      "Experimental::"
+      "HIP, Kokkos::Experimental::HIPSpace>, "
+      "LAMMPS_NS::PairReaxFFKokkos<Kokkos::Experimental::HIP>::params_sing, false>, "
+      "Kokkos::RangePolicy<Kokkos::Experimental::HIP, Kokkos::IndexType<long> >, "
+      "Kokkos::Experimental::HIP> const*)";
+
+  std::string trunkated_name = rocprofiler::truncate_name(long_kernel_name);
+  EXPECT_EQ("hip_parallel_launch_local_memory", trunkated_name);
 }
\ No newline at end of file