1a96570621
AMDGPU plugin equivalent of D95155, build without HSA installed locally Compiles a new file, plugins/amdgpu/dynamic_hsa/hsa.cpp, to an object file that exposes the same symbols that the plugin presently uses from hsa. The object file contains dlopen of hsa and cached dlsym calls. Also provides header files corresponding to the subset that is used. This is behind a feature flag, LIBOMPTARGET_FORCE_DLOPEN_LIBHSA, default off. That allows developers to build against the dlopen/dlsym implementation, e.g. while testing this mode. Enabling by default will cause this plugin to build on a wider variety of machines than it does at present so may break some CI builds. That risk can be minimised by reviewing the header dependencies of the library and ensuring it doesn't use any libraries that are not already used by libomptarget. Separating the implementation from enabling by default in case the latter needs to be rolled back after wider CI results. Reviewed By: jdoerfert Differential Revision: https://reviews.llvm.org/D106559
95 lines
2.7 KiB
C++
95 lines
2.7 KiB
C++
//===--- amdgpu/dynamic_hsa/hsa.cpp ------------------------------- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Implement subset of hsa api by calling into hsa library via dlopen
|
|
// Does the dlopen/dlsym calls as part of the call to hsa_init
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#include "hsa.h"
|
|
#include "Debug.h"
|
|
#include "dlwrap.h"
|
|
#include "hsa_ext_amd.h"
|
|
|
|
#include <dlfcn.h>
|
|
|
|
DLWRAP_INTERNAL(hsa_init, 0);
|
|
|
|
DLWRAP(hsa_status_string, 2);
|
|
DLWRAP(hsa_shut_down, 0);
|
|
DLWRAP(hsa_agent_get_info, 3);
|
|
DLWRAP(hsa_iterate_agents, 2);
|
|
DLWRAP(hsa_signal_create, 4);
|
|
DLWRAP(hsa_signal_destroy, 1);
|
|
DLWRAP(hsa_signal_store_relaxed, 2);
|
|
DLWRAP(hsa_signal_store_screlease, 2);
|
|
DLWRAP(hsa_signal_wait_scacquire, 5);
|
|
DLWRAP(hsa_queue_create, 8);
|
|
DLWRAP(hsa_queue_load_read_index_scacquire, 1);
|
|
DLWRAP(hsa_queue_add_write_index_relaxed, 2);
|
|
DLWRAP(hsa_memory_copy, 3);
|
|
DLWRAP(hsa_executable_create, 4);
|
|
DLWRAP(hsa_executable_destroy, 1);
|
|
DLWRAP(hsa_executable_freeze, 2);
|
|
DLWRAP(hsa_executable_symbol_get_info, 3);
|
|
DLWRAP(hsa_executable_iterate_symbols, 3);
|
|
DLWRAP(hsa_code_object_deserialize, 4);
|
|
DLWRAP(hsa_executable_load_code_object, 4);
|
|
DLWRAP(hsa_amd_agent_memory_pool_get_info, 4);
|
|
DLWRAP(hsa_amd_agent_iterate_memory_pools, 3);
|
|
DLWRAP(hsa_amd_memory_pool_allocate, 4);
|
|
DLWRAP(hsa_amd_memory_pool_free, 1);
|
|
DLWRAP(hsa_amd_memory_async_copy, 8);
|
|
DLWRAP(hsa_amd_memory_pool_get_info, 3);
|
|
DLWRAP(hsa_amd_agents_allow_access, 4);
|
|
DLWRAP(hsa_amd_memory_fill, 3);
|
|
DLWRAP(hsa_amd_register_system_event_handler, 2);
|
|
|
|
DLWRAP_FINALIZE();
|
|
|
|
#ifndef DYNAMIC_HSA_PATH
|
|
#define DYNAMIC_HSA_PATH "libhsa-runtime64.so"
|
|
#endif
|
|
|
|
#ifndef TARGET_NAME
|
|
#define TARGET_NAME AMDHSA
|
|
#endif
|
|
#define DEBUG_PREFIX "Target " GETNAME(TARGET_NAME) " RTL"
|
|
|
|
static bool checkForHSA() {
|
|
// return true if dlopen succeeded and all functions found
|
|
|
|
const char *HsaLib = DYNAMIC_HSA_PATH;
|
|
void *DynlibHandle = dlopen(HsaLib, RTLD_NOW);
|
|
if (!DynlibHandle) {
|
|
DP("Unable to load library '%s': %s!\n", HsaLib, dlerror());
|
|
return false;
|
|
}
|
|
|
|
for (size_t I = 0; I < dlwrap::size(); I++) {
|
|
const char *Sym = dlwrap::symbol(I);
|
|
|
|
void *P = dlsym(DynlibHandle, Sym);
|
|
if (P == nullptr) {
|
|
DP("Unable to find '%s' in '%s'!\n", Sym, HsaLib);
|
|
return false;
|
|
}
|
|
DP("Implementing %s with dlsym(%s) -> %p\n", Sym, Sym, P);
|
|
|
|
*dlwrap::pointer(I) = P;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
hsa_status_t hsa_init() {
|
|
if (!checkForHSA()) {
|
|
return HSA_STATUS_ERROR;
|
|
}
|
|
return dlwrap_hsa_init();
|
|
}
|