Make COMPILED/EAGER_ITERATION_TIMES local + bugfix print_time_stats

iree-org / iree-torch

Torch Frontend for IREE

Apache License 2.0

25 stars 11 forks source link

Make COMPILED/EAGER_ITERATION_TIMES local + bugfix print_time_stats #30

Closed ramiro050 closed 2 years ago

ramiro050 commented 2 years ago

A previous commit introduced a bug in print_time_stats by removing the line that turns the input list into a torch tensor. This commit adds that back in.