ROC profiler prototype sources importing

[ROCm/rocprofiler commit: 85278f08a0]
2017-11-09 17:26:19 -06:00
parent b73a5703b5
commit 5fea997bc9
63 changed files with 7598 additions and 0 deletions
@@ -0,0 +1,70 @@
+#ifndef _SRC_CORE_PROXY_QUEUE_H
+#define _SRC_CORE_PROXY_QUEUE_H
+
+#include <hsa.h>
+#include <hsa_api_trace.h>
+#include <atomic>
+#include <map>
+#include <mutex>
+
+#include "core/queue.h"
+#include "core/types.h"
+
+struct HsaApiTable;
+
+namespace rocprofiler {
+typedef void (*hsa_amd_queue_intercept_packet_writer)(const void* packets, uint64_t count);
+typedef void (*on_submit_cb_t)(const void* packet, uint64_t count, uint64_t que_idx, void* data, hsa_amd_queue_intercept_packet_writer writer);
+
+class ProxyQueue : public Queue {
+  public:
+  static void InitFactory() {
+#ifdef ROCP_HSA_PROXY
+    const char* type = getenv("ROCP_PROXY_QUEUE");
+    if (type != NULL) {
+      if (strncmp(type, "rocp", 4) == 0) rocp_type_ = true;
+    }
+#else
+    rocp_type_ = true;
+#endif
+  }
+
+  static void HsaIntercept(HsaApiTable* table);
+
+  static ProxyQueue* Create(
+    hsa_agent_t agent,
+    uint32_t size,
+    hsa_queue_type32_t type,
+    void (*callback)(hsa_status_t status, hsa_queue_t *source, void *data),
+    void *data,
+    uint32_t private_segment_size,
+    uint32_t group_segment_size,
+    hsa_queue_t **queue,
+    hsa_status_t* status);
+
+  static hsa_status_t Destroy(const ProxyQueue* obj);
+
+  virtual hsa_status_t Init(
+    hsa_agent_t agent,
+    uint32_t size,
+    hsa_queue_type32_t type,
+    void (*callback)(hsa_status_t status, hsa_queue_t *source, void *data),
+    void *data,
+    uint32_t private_segment_size,
+    uint32_t group_segment_size,
+    hsa_queue_t **queue
+  ) = 0;
+  virtual hsa_status_t Cleanup() const = 0;
+  virtual hsa_status_t SetInterceptCB(on_submit_cb_t on_submit_cb, void* data) = 0;
+  virtual void Submit(const packet_t* packet) = 0;
+
+  protected:
+  virtual ~ProxyQueue() {};
+
+  private:
+  static bool rocp_type_;
+};
+
+} // namespace rocprofiler
+
+#endif // _SRC_CORE_PROXY_QUEUE_H