Initial set of changes for ThreadTrace

Change-Id: I07ce31f9b4f508cef0fc9ca6dadcf26b6c90361e [ROCm/ROCR-Runtime commit: eb2efb83d1]
2016-10-05 00:15:13 -05:00
parent 05a7b5a9e4
commit e3d8eac639
8 changed files with 86 additions and 6 deletions
@@ -915,6 +915,25 @@ size_t ExecutableImpl::QuerySegmentDescriptors(
  return i - first_empty_segment_descriptor;
 }

+hsa_executable_t AmdHsaCodeLoader::FindExecutable(uint64_t device_address)
+{
+  hsa_executable_t execHandle = {0};
+  ReaderLockGuard<ReaderWriterLock> reader_lock(rw_lock_);
+  if (device_address == 0) {
+    return execHandle;
+  }
+
+  for (auto &exec : executables) {
+    if (exec != nullptr) {
+      uint64_t host_address = exec->FindHostAddress(device_address);
+      if (host_address != 0) {
+        return Executable::Handle(exec);
+      }
+    }
+  }
+  return execHandle;
+}
+
 uint64_t ExecutableImpl::FindHostAddress(uint64_t device_address)
 {
  for (auto &obj : loaded_code_objects) {