merrymercy opened a new pull request #7166: URL: https://github.com/apache/tvm/pull/7166
Currently, due to the conflict between python' multiprocessing and tvm's thread pool, we cannot run auto-scheduler search after running any tvm-generated functions. Luckily, I found this conflict can be avoided if we run the measurement function in `LocalRunner` within a new `threading.Thread`. This PR just added such a new wrapper in `auto_scheduler.local_runner.run`. With this PR, we can freely run auto-scheduler search after executing tvm-generated functions or relay's FoldConstant. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
