HSA API callbacks and hsa test

inc/ext/hsa_rt_utils.hpp

@@ -27,6 +27,7 @@ THE SOFTWARE.
 
 #include <cstdint>
 #include <cstddef>
+#include <iostream>
 
 #define HSART_CALL(call)                                                                           \
   do {                                                                                             \
@@ -45,10 +46,13 @@ class Timer {
   public:
   typedef uint64_t timestamp_t;
   typedef long double freq_t;
+  typedef decltype(hsa_system_get_info)* hsa_system_get_info_fn_t;
 
-  Timer() {
+  // Initialization
+  inline void init(const hsa_system_get_info_fn_t& get_info_fn) {
+    hsa_system_get_info_fn = get_info_fn;
     timestamp_t timestamp_hz = 0;
-    HSART_CALL(hsa_system_get_info(HSA_SYSTEM_INFO_TIMESTAMP_FREQUENCY, &timestamp_hz));
+    HSART_CALL(get_info_fn(HSA_SYSTEM_INFO_TIMESTAMP_FREQUENCY, &timestamp_hz));
     timestamp_rate_ = (freq_t)1000000000 / (freq_t)timestamp_hz;
   }
 
@@ -66,8 +70,23 @@ class Timer {
     HSART_CALL(hsa_system_get_info(HSA_SYSTEM_INFO_TIMESTAMP, &timestamp));
     return timestamp_to_ns(timestamp);
   }
+  timestamp_t timestamp_fn_ns() const {
+    timestamp_t timestamp;
+    HSART_CALL(hsa_system_get_info_fn(HSA_SYSTEM_INFO_TIMESTAMP, &timestamp));
+    return timestamp_to_ns(timestamp);
+  }
+
+  Timer() {
+    init(hsa_system_get_info);
+  }
+
+  Timer(hsa_system_get_info_fn_t f) {
+    init(f);
+  }
 
   private:
+  // hsa_system_get_info function
+  hsa_system_get_info_fn_t hsa_system_get_info_fn;
   // Timestamp rate
   freq_t timestamp_rate_;
 };

inc/ext/prof_protocol.h

@@ -26,9 +26,10 @@ THE SOFTWARE.
 // Traced API domains
 typedef enum {
   ACTIVITY_DOMAIN_ANY = 0,                        // Any domain
-  ACTIVITY_DOMAIN_HIP_API = 1,                    // HIP domain
+  ACTIVITY_DOMAIN_HSA_API = 1,                    // HSA domain
   ACTIVITY_DOMAIN_HCC_OPS = 2,                    // HCC domain
-  ACTIVITY_DOMAIN_NUMBER = 3
+  ACTIVITY_DOMAIN_HIP_API = 3,                    // HIP domain
+  ACTIVITY_DOMAIN_NUMBER = 4
 } activity_domain_t;
 
 // API calback type