From 7cb973d1e14ebe0c2a68e9f2035b34436de4ffcb Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Tue, 9 Jan 2024 15:27:07 -0800 Subject: [PATCH] [Unity] Fix creation of disco ProcessSession --- python/tvm/runtime/disco/session.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/runtime/disco/session.py b/python/tvm/runtime/disco/session.py index 8d0ff57a32bf..b166bd82e9e5 100644 --- a/python/tvm/runtime/disco/session.py +++ b/python/tvm/runtime/disco/session.py @@ -364,8 +364,8 @@ def __init__(self, num_workers: int, entrypoint: str) -> None: self.__init_handle_by_constructor__( _ffi_api.SessionProcess, # type: ignore # pylint: disable=no-member num_workers, - entrypoint, "runtime.disco.create_process_pool", + entrypoint, )