Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[BUG] TVM PopenPoolExecutor may have some bugs on TL Scripts #211

Open
2 tasks
LeiWang1999 opened this issue Oct 2, 2024 · 3 comments
Open
2 tasks

[BUG] TVM PopenPoolExecutor may have some bugs on TL Scripts #211

LeiWang1999 opened this issue Oct 2, 2024 · 3 comments
Assignees
Labels
enhancement New feature or request good first issue Good for newcomers

Comments

@LeiWang1999
Copy link
Contributor

LeiWang1999 commented Oct 2, 2024

When using PopenPoolExecutor for parallel lowering in TLScript, an unexpected bug occurs. An IRModule that can be successfully lowered outside of PopenPoolExecutor fails to lower after being passed through PopenPoolExecutor. The issue seems to be related to serialization and deserialization in PopenPoolExecutor.

This problem occurs even though the script itself appears correct, and using the same IRModule outside of PopenPoolExecutor works as expected. The executor in use is tvm’s custom implementation of PopenPoolExecutor.

Tasks

Preview Give feedback
@LeiWang1999
Copy link
Contributor Author

tvm.contrib.popen_pool.PopenPoolExecutor perfectly works on packing and re-packing TIR/TE Related items. But TL may introduce some extra stuff that breaks the pipeline.

@LeiWang1999
Copy link
Contributor Author

Closed, PR #207 has been updated into ThreadPoolExecutor as the workload is not too heavy.

@LeiWang1999
Copy link
Contributor Author

The problem may be due to the missing implementation for the IRPrinter and Parser of T.Layout.

@LeiWang1999 LeiWang1999 reopened this Oct 3, 2024
@LeiWang1999 LeiWang1999 self-assigned this Oct 15, 2024
@LeiWang1999 LeiWang1999 added enhancement New feature or request good first issue Good for newcomers labels Dec 16, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

1 participant