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

Query DMABuf support through HSA runtime API (#654)

Browse Source
This commit is contained in:
Wenkai Du
2022-11-30 08:53:03 -08:00
committed by GitHub
parent 9594bbee3b
commit fb9938cffa
2 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/misc/rocmwrap.cc
+10 -4
View File
@@ -96,11 +96,17 @@ ncclResult_t rocmLibraryInit(void) {
//goto error;
//}
pfn_hsa_amd_portable_export_dmabuf = (PFN_hsa_amd_portable_export_dmabuf) dlsym(hsaLib, "hsa_amd_portable_export_dmabuf");
if (pfn_hsa_amd_portable_export_dmabuf == NULL) {
WARN("Failed to load ROCr missing symbol hsa_amd_portable_export_dmabuf");
goto error;
#if defined(QUERY_HSA_DMABUF)
bool supported;
res = pfn_hsa_system_get_info(HSA_AMD_SYSTEM_INFO_DMABUF_SUPPORTED, &supported);
if (res == 0 && supported) {
pfn_hsa_amd_portable_export_dmabuf = (PFN_hsa_amd_portable_export_dmabuf) dlsym(hsaLib, "hsa_amd_portable_export_dmabuf");
if (pfn_hsa_amd_portable_export_dmabuf == NULL) {
WARN("Failed to load ROCr missing symbol hsa_amd_portable_export_dmabuf");
goto error;
}
}
#endif
/*
* Required to initialize the ROCr Driver.
* Multiple calls of hsa_init() will return immediately