1 //===--- amdgpu/dynamic_hsa/hsa.cpp ------------------------------- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Implement subset of hsa api by calling into hsa library via dlopen
10 // Does the dlopen/dlsym calls as part of the call to hsa_init
11 //
12 //===----------------------------------------------------------------------===//
13 #include "hsa.h"
14 #include "Debug.h"
15 #include "dlwrap.h"
16 #include "hsa_ext_amd.h"
17 
18 #include <dlfcn.h>
19 
20 DLWRAP_INTERNAL(hsa_init, 0);
21 
22 DLWRAP(hsa_status_string, 2);
23 DLWRAP(hsa_shut_down, 0);
24 DLWRAP(hsa_agent_get_info, 3);
25 DLWRAP(hsa_iterate_agents, 2);
26 DLWRAP(hsa_signal_create, 4);
27 DLWRAP(hsa_signal_destroy, 1);
28 DLWRAP(hsa_signal_store_relaxed, 2);
29 DLWRAP(hsa_signal_store_screlease, 2);
30 DLWRAP(hsa_signal_wait_scacquire, 5);
31 DLWRAP(hsa_queue_create, 8);
32 DLWRAP(hsa_queue_load_read_index_scacquire, 1);
33 DLWRAP(hsa_queue_add_write_index_relaxed, 2);
34 DLWRAP(hsa_memory_copy, 3);
35 DLWRAP(hsa_executable_create, 4);
36 DLWRAP(hsa_executable_destroy, 1);
37 DLWRAP(hsa_executable_freeze, 2);
38 DLWRAP(hsa_executable_symbol_get_info, 3);
39 DLWRAP(hsa_executable_iterate_symbols, 3);
40 DLWRAP(hsa_code_object_deserialize, 4);
41 DLWRAP(hsa_executable_load_code_object, 4);
42 DLWRAP(hsa_amd_agent_memory_pool_get_info, 4);
43 DLWRAP(hsa_amd_agent_iterate_memory_pools, 3);
44 DLWRAP(hsa_amd_memory_pool_allocate, 4);
45 DLWRAP(hsa_amd_memory_pool_free, 1);
46 DLWRAP(hsa_amd_memory_async_copy, 8);
47 DLWRAP(hsa_amd_memory_pool_get_info, 3);
48 DLWRAP(hsa_amd_agents_allow_access, 4);
49 DLWRAP(hsa_amd_memory_fill, 3);
50 DLWRAP(hsa_amd_register_system_event_handler, 2);
51 
52 DLWRAP_FINALIZE();
53 
54 #ifndef DYNAMIC_HSA_PATH
55 #define DYNAMIC_HSA_PATH "libhsa-runtime64.so"
56 #endif
57 
58 #ifndef TARGET_NAME
59 #error "Missing TARGET_NAME macro"
60 #endif
61 #define DEBUG_PREFIX "Target " GETNAME(TARGET_NAME) " RTL"
62 
63 static bool checkForHSA() {
64   // return true if dlopen succeeded and all functions found
65 
66   const char *HsaLib = DYNAMIC_HSA_PATH;
67   void *DynlibHandle = dlopen(HsaLib, RTLD_NOW);
68   if (!DynlibHandle) {
69     DP("Unable to load library '%s': %s!\n", HsaLib, dlerror());
70     return false;
71   }
72 
73   for (size_t I = 0; I < dlwrap::size(); I++) {
74     const char *Sym = dlwrap::symbol(I);
75 
76     void *P = dlsym(DynlibHandle, Sym);
77     if (P == nullptr) {
78       DP("Unable to find '%s' in '%s'!\n", Sym, HsaLib);
79       return false;
80     }
81     DP("Implementing %s with dlsym(%s) -> %p\n", Sym, Sym, P);
82 
83     *dlwrap::pointer(I) = P;
84   }
85 
86   return true;
87 }
88 
89 hsa_status_t hsa_init() {
90   if (!checkForHSA()) {
91     return HSA_STATUS_ERROR;
92   }
93   return dlwrap_hsa_init();
94 }
95