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

vinx13 · web-flow · commit 81a6c51ba433 · 2024-01-11T17:26:02.000-08:00
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,
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -364,8 +364,8 @@ def __init__(self, num_workers: int, entrypoint: str) -> None:`
`364`	`364`	`self.__init_handle_by_constructor__(`
`365`	`365`	`_ffi_api.SessionProcess, # type: ignore # pylint: disable=no-member`
`366`	`366`	`num_workers,`
`367`		`- entrypoint,`
`368`	`367`	`"runtime.disco.create_process_pool",`
	`368`	`+ entrypoint,`
`369`	`369`	`)`
`370`	`370`
`371`	`371`