AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-546285 - add hipGetDriverEntryPoint (#855)

Browse Source 
[ROCm/clr commit: 789e2029ca]
This commit is contained in:
Li, Todd tiantuo
2025-08-15 20:08:21 -07:00
committed by GitHub
parent 405061b019
commit ad9eb56dd4
9 changed files with 134 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_device_runtime.cpp
+55
View File
@@ -21,6 +21,7 @@
#include <hip/hip_runtime.h>
#include "hip_internal.hpp"
#include "hip_platform.hpp"
#undef hipChooseDevice
#undef hipDeviceProp_t
@@ -716,6 +717,60 @@ hipError_t hipGetDeviceFlags(unsigned int* flags) {
HIP_RETURN(hipSuccess);
}
hipError_t hipGetDriverEntryPoint_common(const char* symbol, void** funcPtr, unsigned long long flags,
hipDriverEntryPointQueryResult* status) {
std::string symbolString = symbol;
if (symbol == nullptr || symbolString == "" || funcPtr == nullptr) {
return hipErrorInvalidValue;
}
if (flags != hipEnableDefault && flags != hipEnableLegacyStream
&& flags != hipEnablePerThreadDefaultStream) {
return hipErrorInvalidValue;
}
void* handle = hip::PlatformState::instance().getDynamicLibraryHandle();
if (handle == nullptr) {
return hipErrorInvalidValue;
}
if (flags == hipEnablePerThreadDefaultStream) {
symbolString += "_spt";
}
*funcPtr = amd::Os::getSymbol(handle, symbolString.c_str());
if (funcPtr == nullptr) {
if (flags == hipEnablePerThreadDefaultStream) {
*funcPtr = amd::Os::getSymbol(handle, symbol);
}
if (funcPtr == nullptr) {
if (status != nullptr) {
*status = hipDriverEntryPointSymbolNotFound;
}
return hipErrorInvalidValue;
}
}
if (status != nullptr) {
*status = hipDriverEntryPointSuccess;
}
return hipSuccess;
}
hipError_t hipGetDriverEntryPoint(const char* symbol, void** funcPtr, unsigned long long flags,
hipDriverEntryPointQueryResult* status) {
HIP_INIT_API(hipGetDriverEntryPoint, symbol, funcPtr, flags, status);
HIP_RETURN(hipGetDriverEntryPoint_common(symbol, funcPtr, flags, status));
}
hipError_t hipGetDriverEntryPoint_spt(const char* symbol, void** funcPtr, unsigned long long flags,
hipDriverEntryPointQueryResult* status) {
HIP_INIT_API(hipGetDriverEntryPoint, symbol, funcPtr, flags, status);
flags = (flags == hipEnableDefault) ? hipEnablePerThreadDefaultStream : flags;
HIP_RETURN(hipGetDriverEntryPoint_common(symbol, funcPtr, flags, status));
}
hipError_t hipSetDevice(int device) {
HIP_INIT_API_NO_RETURN(hipSetDevice, device);