Skip to content

GitLab

Commit aa05dfd5 authored by zhuwenwen's avatar zhuwenwen
Browse files

[Fix] nccl暂不支持ncclCommWindowDeregister接口，引入nccl.so时报错问题解决

parent 56983e67
Show whitespace changes
Inline Side-by-side
Showing with 12 additions and 12 deletions
vllm/distributed/device_communicators/pynccl_wrapper.py
View file @ aa05dfd5
...@@ -281,20 +281,20 @@ class NCCLLibrary: ...@@ -281,20 +281,20 @@ class NCCLLibrary:
# ncclResult_t ncclCommWindowRegister( # ncclResult_t ncclCommWindowRegister(
# ncclComm_t comm, void* buff, size_t size, # ncclComm_t comm, void* buff, size_t size,
# ncclWindow_t* win, int winFlags); # ncclWindow_t* win, int winFlags);
Function( # Function(
"ncclCommWindowRegister", # "ncclCommWindowRegister",
ncclResult_t, # ncclResult_t,
[ # [
ncclComm_t, # ncclComm_t,
buffer_type, # buffer_type,
ctypes.c_size_t, # ctypes.c_size_t,
ctypes.POINTER(ncclWindow_t), # ctypes.POINTER(ncclWindow_t),
ctypes.c_int, # ctypes.c_int,
], # ],
), # ),
# ncclResult_t ncclCommWindowDeregister( # ncclResult_t ncclCommWindowDeregister(
# ncclComm_t comm, ncclWindow_t win); # ncclComm_t comm, ncclWindow_t win);
Function("ncclCommWindowDeregister", ncclResult_t, [ncclComm_t, ncclWindow_t]), # Function("ncclCommWindowDeregister", ncclResult_t, [ncclComm_t, ncclWindow_t]),
] ]
# class attribute to store the mapping from the path to the library # class attribute to store the mapping from the path to the library
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment