Skip to content

Commit

Permalink
[Bugfix][MetaSchedule] RPC port needs to be an integer (#14781)
Browse files Browse the repository at this point in the history
This PR introduces a bugfix where the RPC port could be mistakenly
detected as a string from environment variables, which is required to be
an integer for subsequent operators. We cast the type to int in
auto detection.
  • Loading branch information
junrushao authored May 6, 2023
1 parent ca7c3d1 commit 298a0a4
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions python/tvm/meta_schedule/runner/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,10 +105,15 @@ def _sanity_check(self) -> None:
def _normalized(config: Optional["RPCConfig"]) -> "RPCConfig":
if config is None:
config = RPCConfig()
tracker_host = config.tracker_host or os.environ.get("TVM_TRACKER_HOST", None)
tracker_port = config.tracker_port or os.environ.get("TVM_TRACKER_PORT", None)
tracker_key = config.tracker_key or os.environ.get("TVM_TRACKER_KEY", None)
if isinstance(tracker_port, str):
tracker_port = int(tracker_port)
config = RPCConfig(
tracker_host=config.tracker_host or os.environ.get("TVM_TRACKER_HOST", None),
tracker_port=config.tracker_port or os.environ.get("TVM_TRACKER_PORT", None),
tracker_key=config.tracker_key or os.environ.get("TVM_TRACKER_KEY", None),
tracker_host=tracker_host,
tracker_port=tracker_port,
tracker_key=tracker_key,
session_priority=config.session_priority,
session_timeout_sec=config.session_timeout_sec,
)
Expand Down

0 comments on commit 298a0a4

Please sign in to comment.