areusch commented on a change in pull request #5417:
URL: https://github.com/apache/incubator-tvm/pull/5417#discussion_r414918275
##########
File path: src/runtime/micro/micro_session.cc
##########
@@ -489,6 +629,16 @@ PackedFunc MicroSession::GetFunction(
return PackedFunc([sptr_to_self](TVMArgs args, TVMRetValue* rv) {
MicroSession::ExitWithScope();
});
+ // TODO(weberlo): add a `clear_batch_timer` func
+ } else if (name == "get_last_batch_time") {
+ return PackedFunc([sptr_to_self, this](TVMArgs args, TVMRetValue* rv) {
+ *rv = this->GetLastBatchTime();
+ });
+ // TODO(weberlo): remove this func
+ } else if (name == "get_last_batch_cycles") {
+ return PackedFunc([sptr_to_self, this](TVMArgs args, TVMRetValue* rv) {
+ *rv = this->GetLastBatchCycles();
+ });
Review comment:
right now it returns either last host or last device time based on
`use_device_timer`. I agree we should rethink the timing API, but perhaps we
can move it to next PR, when I would want to implement an API between the host
and device, and we can better define the concept of batch time (and the units
it is logged in) there?
----------------------------------------------------------------
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:
us...@infra.apache.org
