"docs/vscode:/vscode.git/clone" did not exist on "60bcef000ebcfaf120edc1972a8136344d9bfa0d"
Commit a14aa44f authored by xuxz's avatar xuxz
Browse files

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

parent 39a5084a
......@@ -226,21 +226,21 @@ class NCCLLibrary:
# ncclResult_t ncclCommWindowRegister(
# ncclComm_t comm, void* buff, size_t size,
# ncclWindow_t* win, int winFlags);
Function(
"ncclCommWindowRegister",
ncclResult_t,
[
ncclComm_t,
buffer_type,
ctypes.c_size_t,
ctypes.POINTER(ncclWindow_t),
ctypes.c_int,
],
),
#Function(
# "ncclCommWindowRegister",
# ncclResult_t,
# [
# ncclComm_t,
# buffer_type,
# ctypes.c_size_t,
# ctypes.POINTER(ncclWindow_t),
# ctypes.c_int,
# ],
#),
# ncclResult_t ncclCommWindowDeregister(
# 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
......
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