[Unity] Fix creation of disco ProcessSession (apache#16375)

tqchen · Jan 12, 2024 · 81a6c51 · 81a6c51
1 parent b8230f6
commit 81a6c51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
         )