wsl/libhsakmt: fix default device in memory allocation

Signed-off-by: Flora Cui <flora.cui@amd.com>
Reviewed-by: Tianci Yin <tianci.yin@amd.com>
Part-of: <http://10.67.69.192/wsl/rocr-runtime/-/merge_requests/82>
Этот коммит содержится в:
Flora Cui
2025-06-30 13:51:12 +08:00
коммит произвёл Frank Min
родитель 67332944d2
Коммит 75cbeda847
2 изменённых файлов: 23 добавлений и 14 удалений
+4 -1
Просмотреть файл
@@ -102,7 +102,10 @@ HSAKMTAPI int amdgpu_bo_import(amdgpu_device_handle dev,
enum amdgpu_bo_handle_type type,
uint32_t shared_handle,
struct amdgpu_bo_import_result *output) {
HsaGraphicsResourceInfo GraphicsResourceInfo;
wsl::thunk::WDDMDevice *pDevice = reinterpret_cast<wsl::thunk::WDDMDevice *>(dev);
HsaGraphicsResourceInfo GraphicsResourceInfo = {};
GraphicsResourceInfo.NodeId = pDevice->NodeId();
HSAKMT_STATUS ret = hsaKmtImportDMABufHandle(shared_handle, &GraphicsResourceInfo);
if (ret == HSAKMT_STATUS_SUCCESS) {
//use GpuMemory object's address as drm buf handle