comaniac commented on a change in pull request #8560:
URL: https://github.com/apache/tvm/pull/8560#discussion_r677158710
##########
File path: python/tvm/auto_scheduler/relay_integration.py
##########
@@ -379,11 +382,37 @@ def auto_schedule_topi(func_name, outs):
@tvm._ffi.register_func("auto_scheduler.relay_integration.te_compiler_update_weights")
def te_compiler_update_weights(function_weights):
- """A callback for updating the weights of extracted tasks."""
+ """A callback for updating the weights of extracted tasks. When using the
TE compiler
+ that avoids compiling the same function multiple times by caching, all
extracted tasks
+ have weight=1, so the TE compiler invokes this callback at the end. In
this case,
Review comment:
```suggestion
have weight 1, so the TE compiler invokes this callback at the end. In
this case,
```
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
