committing kp sampler with fix to scan

kokkos · Jul 31, 2023 · 3254951 · 3254951
1 parent d7aa5bc
commit 3254951
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/common/kokkos-sampler/kp_sampler_skip.cpp b/common/kokkos-sampler/kp_sampler_skip.cpp
@@ -277,12 +277,13 @@ void kokkosp_begin_parallel_reduce(const char* name, const uint32_t devID,
   if ((invocationNum % kernelSampleSkip) == 0) {
     if ((rand() / RAND_MAX) < tool_prob_num / 100.0) {
       if (tool_verbosity > 0) {
-        printf(
+        *kID = 1;
+	 printf(
             "KokkosP: sample %llu (a parallel_reduce on its invocation number "
             "%d) calling child-begin function...\n",
             (unsigned long long)(*kID), (int)invocationNum);
       }
-      *kID = 1;
+
       if (NULL != beginReduceCallee) {
         (*beginReduceCallee)(name, devID, kID);
       }