Backend level argument for iprof

utils/babeltrace_thapi.in

@@ -24,7 +24,8 @@ $options_tally = {
   'display_mode' => [String, 'human'],
   'display_metadata' => [FalseClass, false],
   'display_name_max_size' => [Integer, 80],
-  'display_kernel_verbose' => [FalseClass, false]
+  'display_kernel_verbose' => [FalseClass, false],
+  'backend_level' => [String, ''],
 }
 
 def common_options(opts)

utils/xprof_utils.cpp

@@ -205,4 +205,3 @@ bt_message* create_traffic_message(const char *hostname, const process_id_t proc
 
      return message;
 }
-

utils/xprof_utils.hpp

@@ -5,23 +5,26 @@
 #include <string>
 #include "babeltrace2/babeltrace.h"
 
-enum backend_e{ BACKEND_UNKNOWN = 0,
-                BACKEND_ZE = 1,
-                BACKEND_OPENCL = 2,
-                BACKEND_CUDA = 3,
-                BACKEND_OMP_TARGET_OPERATIONS = 4,
-                BACKEND_OMP = 5,
-                BACKEND_HIP = 6 };
-
-constexpr int backend_level[] = { 2, 2, 2, 2, 1, 0, 2 };
-
-constexpr const char* backend_name[] = { "BACKEND_UNKNOWN",
-                "BACKEND_ZE",
-                "BACKEND_OPENCL",
-                "BACKEND_CUDA",
-                "BACKEND_OMP_TARGET_OPERATIONS",
-                "BACKEND_OMP",
-                "BACKEND_HIP" };
+enum backend_e {
+  BACKEND_UNKNOWN = 0,
+  BACKEND_ZE = 1,
+  BACKEND_OPENCL = 2,
+  BACKEND_CUDA = 3,
+  BACKEND_OMP_TARGET_OPERATIONS = 4,
+  BACKEND_OMP = 5,
+  BACKEND_HIP = 6,
+  BACKEND_MAX,
+};
+
+constexpr const char* backend_name[] = {
+  "BACKEND_UNKNOWN",
+  "BACKEND_ZE",
+  "BACKEND_OPENCL",
+  "BACKEND_CUDA",
+  "BACKEND_OMP_TARGET_OPERATIONS",
+  "BACKEND_OMP",
+  "BACKEND_HIP",
+};
 
 typedef enum        backend_e backend_t;
 typedef unsigned    backend_level_t;

xprof/btx_tally_params.yaml

@@ -12,3 +12,5 @@
     :type: integer_unsigned
   - :name: display_kernel_verbose
     :type: bool
+  - :name: backend_level
+    :type: string

xprof/tally.cpp

@@ -1,4 +1,8 @@
 #include "tally.hpp"
+#include <string>
+#include <array>
+#include <cassert>
+#include <sstream>      // std::stringstream, std::stringbuf
 
 //! User data collection structure.
 //! It is used to collect interval messages data, once data is collected,
@@ -7,6 +11,8 @@ struct tally_dispatch_s {
   //! User params provided to the user component.
   btx_params_t *params;
 
+  std::array<int, BACKEND_MAX>backend_level;
+
   std::map<backend_level_t, std::set<const char *>> host_backend_name;
   std::map<backend_level_t, std::set<const char *>> traffic_backend_name;
 
@@ -32,14 +38,46 @@ void print_metadata(std::vector<std::string> metadata) {
     std::cout << value << std::endl;
 }
 
+int get_backend_id(std::string name) {
+  for(int i = 0; i < BACKEND_MAX; ++i)
+    // backend_name is located in xprof_utils.cpp
+    if (std::string{backend_name[i]} == name) return i;
+  return -1;
+}
+
 void btx_initialize_usr_data(void *btx_handle, void **usr_data) {
   /* User allocates its own data structure */
-  *usr_data = new tally_dispatch_t;
+  auto *data = new tally_dispatch_t;
+  *usr_data = data;
+
+  /* Backend information must match enum backend_e in xprof_utils.hpp */
+  data->backend_level = {
+    2, // BACKEND_UNKNOWN
+    2, // BACKEND_ZE
+    2, // BACKEND_OPENCL
+    2, // BACKEND_CUDA
+    1, // BACKEND_OMP_TARGET_OPERATIONS
+    0, // BACKEND_OMP
+    2, // BACKEND_HIP
+  };
 }
 
 void btx_read_params(void *btx_handle, void *usr_data, btx_params_t *usr_params) {
   tally_dispatch_t *data = (tally_dispatch_t *)usr_data;
   data->params = usr_params;
+
+  // Consumes key:value pairs in the stringstream k1:v1,..,kn:vn
+  std::stringstream tokens{data->params->backend_level};
+  std::string tmp;
+  while (std::getline(tokens, tmp, ',')) {
+    std::stringstream tmp_string{tmp};
+    std::string k,v;
+    std::getline(tmp_string, k, ':');
+    int id = get_backend_id(k);
+    assert((id > 0) && "Backend not found. Please check --backend-level format.");
+    std::getline(tmp_string, v);
+    data->backend_level[id] = std::stoi(v);
+  }
 }
 
 void btx_finalize_usr_data(void *btx_handle, void *usr_data) {
@@ -127,7 +165,7 @@ static void host_usr_callback(void *btx_handle, void *usr_data, const char *host
   tally_dispatch_t *data = (tally_dispatch_t *)usr_data;
 
   TallyCoreTime a{dur, (uint64_t)err};
-  const int level = backend_level[backend_id];
+  const int level = data->backend_level[backend_id];
   data->host_backend_name[level].insert(backend_name[backend_id]);
   data->host[level][hpt_function_name_t(hostname, vpid, vtid, name)] += a;
 }
@@ -154,7 +192,7 @@ static void traffic_usr_callback(void *btx_handle, void *usr_data, const char *h
   tally_dispatch_t *data = (tally_dispatch_t *)usr_data;
 
   TallyCoreByte a{(uint64_t)size, false};
-  const int level = backend_level[backend];
+  const int level = data->backend_level[backend];
   data->traffic_backend_name[level].insert(backend_name[backend]);
   data->traffic[level][hpt_function_name_t(hostname, vpid, vtid, name)] += a;
 }

xprof/tally.hpp

@@ -9,6 +9,8 @@
 #include <tuple>
 #include <unordered_map>
 #include <vector>
+#include <string>
+#include <array>
 
 #include "json.hpp"
 #include "my_demangle.h"

xprof/xprof.sh.erb.in

@@ -521,6 +521,7 @@ while (( "$#" )); do
         -l | --timeline)        shift; timeline=true ;;
         -j | --json)            shift; bt_tally_argv+=" --display_mode=json" ;;
         -m | --tracing-mode)    shift; tracing_mode=$1; shift ;;
+        --backend-level)        shift; bt_tally_argv+=" --backend_level=$1"; shift ;;
         --no-save)              shift; procesing_mode="on-the-fly" ;;
         --traced-ranks)         shift; traced_ranks=$1; shift ;;
         --asm)                  shift; asm=true ;;