Fixing bug in determining tree node for kokkos tuning

UO-OACISS · Sep 18, 2024 · 8354e6d · 8354e6d
1 parent 5c30039
commit 8354e6d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/apex/apex_kokkos_tuning.cpp b/src/apex/apex_kokkos_tuning.cpp
@@ -1211,10 +1211,12 @@ void kokkosp_request_values(
     Kokkos_Tools_VariableValue* tuningVariableValues) {
     if (!apex::apex_options::use_kokkos_tuning()) { return; }
     // first, get the current timer node in the task tree
-    auto tlt = apex::thread_instance::get_top_level_timer();
+    //auto tlt = apex::thread_instance::get_top_level_timer();
+    auto tlt = apex::thread_instance::get_current_profiler();
     std::string tree_node{"default"};
     if (tlt != nullptr) {
-        tree_node = tlt->tree_node->getName();
+        //tree_node = tlt->tt_ptr->tree_node->getName();
+        tree_node = tlt->tt_ptr->task_id->get_name();
     }
     // don't track memory in this function.
     apex::in_apex prevent_memory_tracking;