Skip to content

[RPC] Fix tuning on macOS and Windows (#15771)#16357

Merged
Johnson9009 merged 2 commits intoapache:mainfrom
gmeeker:bugfix/autotune
Jan 12, 2024
Merged

[RPC] Fix tuning on macOS and Windows (#15771)#16357
Johnson9009 merged 2 commits intoapache:mainfrom
gmeeker:bugfix/autotune

Conversation

@gmeeker
Copy link
Copy Markdown
Contributor

@gmeeker gmeeker commented Jan 6, 2024

Fix regression in (#15187) when multiprocessing start method is not 'fork', which prevented tuning from working. This affects macOS and Windows. Also in python 3.14 the default start method will be 'spawn'.

Fix regression in (apache#15187) when multiprocessing start method is not 'fork',
which prevented tuning from working. This affects macOS and Windows.
Also in python 3.14 the default start method will be 'spawn'.
@junrushao
Copy link
Copy Markdown
Member

CC @Johnson9009

@tqchen tqchen requested a review from Johnson9009 January 11, 2024 00:07
Copy link
Copy Markdown
Contributor

@Johnson9009 Johnson9009 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

sorry for late review, threre are some questions.

Comment thread python/tvm/rpc/server.py Outdated
Comment thread python/tvm/rpc/server.py Outdated
"""Server loop"""
sockfd = sock.fileno()
temp = _server_env(load_library, work_path)
_ffi_api.ServerLoop(sockfd)
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can the sockfd be inline?

Comment thread python/tvm/rpc/server.py Outdated
Comment thread python/tvm/rpc/server.py Outdated


def _serve_loop(sock, load_library, work_path=None):
"""Server loop"""
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

del the meaningless comment because it give no more information beyond the function name, or write a meaning one.

Comment thread python/tvm/rpc/server.py Outdated
Comment thread python/tvm/rpc/server.py Outdated
@gmeeker
Copy link
Copy Markdown
Contributor Author

gmeeker commented Jan 11, 2024

The changes from the original code are now minimal, just removing the closure.

Copy link
Copy Markdown
Contributor

@Johnson9009 Johnson9009 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, look good to me.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants