This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new ccc09fa [TVMC] Switch profile flag to use new profiler (#8710)
ccc09fa is described below
commit ccc09fa7cda5a3975f064d00725aaa619d0c118c
Author: Tristan Konolige <[email protected]>
AuthorDate: Thu Aug 12 21:36:16 2021 -0700
[TVMC] Switch profile flag to use new profiler (#8710)
---
python/tvm/driver/tvmc/model.py | 2 +-
python/tvm/driver/tvmc/runner.py | 4 +++-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/python/tvm/driver/tvmc/model.py b/python/tvm/driver/tvmc/model.py
index 7dc3fd4..a9516e1 100644
--- a/python/tvm/driver/tvmc/model.py
+++ b/python/tvm/driver/tvmc/model.py
@@ -371,7 +371,7 @@ class TVMCPackage(object):
class TVMCResult(object):
"""A class that stores the results of tvmc.run and provides helper
utilities."""
- def __init__(self, outputs: Dict[str, np.ndarray], times: List[str]):
+ def __init__(self, outputs: Dict[str, np.ndarray], times: List[float]):
"""Create a convenience wrapper around the output of tvmc.run
Parameters
diff --git a/python/tvm/driver/tvmc/runner.py b/python/tvm/driver/tvmc/runner.py
index 9161398..8515bc9 100644
--- a/python/tvm/driver/tvmc/runner.py
+++ b/python/tvm/driver/tvmc/runner.py
@@ -417,7 +417,9 @@ def run_module(
# Run must be called explicitly if profiling
if profile:
logger.info("Running the module with profiling enabled.")
- module.run()
+ report = module.profile()
+ # This print is intentional
+ print(report)
# create the module time evaluator (returns a function)
timer = module.module.time_evaluator("run", dev, number=number,
repeat=repeat)
