jcf94 commented on a change in pull request #7877:
URL: https://github.com/apache/tvm/pull/7877#discussion_r615504135
##########
File path: include/tvm/auto_scheduler/measure.h
##########
@@ -340,9 +351,14 @@ class LocalBuilderNode : public ProgramBuilderNode {
public:
/*! \brief Build function. */
String build_func;
+ /*! \brief Functor with python implementation of submit method. */
+ PackedFunc submit_func;
Review comment:
How about to pass a wrapped build function as the build_func like:
```python
wrapped_build_func(original_build_func, ...) {
....
pass the original_build_func to your submit logic
....
}
```
Then we can avoid to add another strange API here.
--
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]
