Skip to content

Commit 3b374b7

Browse files
committed
support gdr for intel xpu
1 parent 14219b1 commit 3b374b7

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/uct/ib/base/ib_md.c

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1341,6 +1341,12 @@ ucs_status_t uct_ib_md_open_common(uct_ib_md_t *md,
13411341
/* check if ROCM KFD driver is loaded */
13421342
uct_ib_check_gpudirect_driver(md, "/dev/kfd", UCS_MEMORY_TYPE_ROCM);
13431343

1344+
#if HAVE_ZE
1345+
uct_ib_check_gpudirect_driver(
1346+
md, "/sys/module/xe/srcversion",
1347+
UCS_MEMORY_TYPE_ZE_DEVICE);
1348+
#endif
1349+
13441350
/* Check for dma-buf support */
13451351
uct_ib_md_check_dmabuf(md);
13461352
}

src/uct/ze/copy/ze_copy_md.c

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ static ucs_status_t uct_ze_copy_md_query(uct_md_h md, uct_md_attr_v2_t *md_attr)
4848
md_attr->alloc_mem_types = UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) |
4949
UCS_BIT(UCS_MEMORY_TYPE_ZE_DEVICE) |
5050
UCS_BIT(UCS_MEMORY_TYPE_ZE_MANAGED);
51-
md_attr->access_mem_types = UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) |
51+
md_attr->access_mem_types = UCS_BIT(UCS_MEMORY_TYPE_HOST) |
52+
UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) |
5253
UCS_BIT(UCS_MEMORY_TYPE_ZE_DEVICE) |
5354
UCS_BIT(UCS_MEMORY_TYPE_ZE_MANAGED);
5455
md_attr->detect_mem_types = UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) |

0 commit comments

Comments
 (0)