kp_sampler_skip.cpp: make probability sampling the priority (as before)

Fix from a previous suggestion from contributed commit - it may have been misunderstood that probability sampling is the priority here (I should have caught this). Co-authored-by: Christian Trott <[email protected]>
kokkos · May 3, 2024 · b03c2c4 · b03c2c4
1 parent bb538bb
commit b03c2c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/common/kokkos-sampler/kp_sampler_skip.cpp b/common/kokkos-sampler/kp_sampler_skip.cpp
@@ -225,7 +225,7 @@ void kokkosp_init_library(const int loadSeq, const uint64_t interfaceVer,
   }
 
   const char* tool_sample = getenv("KOKKOS_TOOLS_SAMPLER_SKIP");
-  if (NULL != tool_sample) {
+  if (NULL != tool_sample && tool_prob_num == -1.0) {
     tool_prob_num    = 100.0;
     kernelSampleSkip = atoi(tool_sample) + 1;
     if (tool_verbosity > 0) {