diff --git a/AUTHORS b/AUTHORS index 9804584832a..f94224ecb69 100644 --- a/AUTHORS +++ b/AUTHORS @@ -109,6 +109,7 @@ Valentin Petrov Wenbin Lu Xin Zhao Xu Yifeng +Yihua Xu Yiltan Hassan Temucin Yossi Itigin Yuriy Shestakov diff --git a/src/uct/ib/base/ib_md.c b/src/uct/ib/base/ib_md.c index 265d53e0a87..ce4f43983ee 100644 --- a/src/uct/ib/base/ib_md.c +++ b/src/uct/ib/base/ib_md.c @@ -1341,6 +1341,12 @@ ucs_status_t uct_ib_md_open_common(uct_ib_md_t *md, /* check if ROCM KFD driver is loaded */ uct_ib_check_gpudirect_driver(md, "/dev/kfd", UCS_MEMORY_TYPE_ROCM); +#if HAVE_ZE + uct_ib_check_gpudirect_driver( + md, "/sys/module/xe/srcversion", + UCS_MEMORY_TYPE_ZE_DEVICE); +#endif + /* Check for dma-buf support */ uct_ib_md_check_dmabuf(md); } @@ -1349,7 +1355,7 @@ ucs_status_t uct_ib_md_open_common(uct_ib_md_t *md, !(md->cap_flags & UCT_MD_FLAG_REG_DMABUF) && (md_config->enable_gpudirect_rdma == UCS_YES)) { ucs_error("%s: Couldn't enable GPUDirect RDMA. Please make sure " - "nv_peer_mem or amdgpu plugin installed correctly, or dmabuf " + "nv_peer_mem, amdgpu plugin, or Intel XE driver is installed correctly, or dmabuf " "is supported.", uct_ib_device_name(&md->dev)); status = UCS_ERR_UNSUPPORTED; diff --git a/src/uct/ze/copy/ze_copy_md.c b/src/uct/ze/copy/ze_copy_md.c index 9a4b2d51268..e70153d2631 100644 --- a/src/uct/ze/copy/ze_copy_md.c +++ b/src/uct/ze/copy/ze_copy_md.c @@ -48,7 +48,8 @@ static ucs_status_t uct_ze_copy_md_query(uct_md_h md, uct_md_attr_v2_t *md_attr) md_attr->alloc_mem_types = UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) | UCS_BIT(UCS_MEMORY_TYPE_ZE_DEVICE) | UCS_BIT(UCS_MEMORY_TYPE_ZE_MANAGED); - md_attr->access_mem_types = UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) | + md_attr->access_mem_types = UCS_BIT(UCS_MEMORY_TYPE_HOST) | + UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) | UCS_BIT(UCS_MEMORY_TYPE_ZE_DEVICE) | UCS_BIT(UCS_MEMORY_TYPE_ZE_MANAGED); md_attr->detect_mem_types = UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) |