Skip to content

GitLab

Unverified Commit 4561c835 authored by guoshzhao's avatar guoshzhao Committed by GitHub
Browse files

specify the framework for matmul example. (#37) 


Co-authored-by: default avatarGuoshuai Zhao <guzhao@microsoft.com>
parent 02eef9ca
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
examples/benchmarks/matmul.py
View file @ 4561c835
...@@ -3,17 +3,16 @@ ...@@ -3,17 +3,16 @@
"""Model benchmark example for matmul with pytorch. """Model benchmark example for matmul with pytorch.
Command to run: Commands to run:
python3 examples/benchmarks/matmul.py python3 examples/benchmarks/matmul.py
""" """
from superbench.benchmarks import BenchmarkRegistry from superbench.benchmarks import Framework, BenchmarkRegistry
from superbench.common.utils import logger from superbench.common.utils import logger
if __name__ == '__main__': if __name__ == '__main__':
context = BenchmarkRegistry.create_benchmark_context( context = BenchmarkRegistry.create_benchmark_context(
'matmul', 'matmul', parameters='--num_steps=20', framework=Framework.PYTORCH
parameters='--num_steps=20',
) )
benchmark = BenchmarkRegistry.launch_benchmark(context) benchmark = BenchmarkRegistry.launch_benchmark(context)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment