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
......@@ -3,17 +3,16 @@
"""Model benchmark example for matmul with pytorch.
Command to run:
Commands to run:
python3 examples/benchmarks/matmul.py
"""
from superbench.benchmarks import BenchmarkRegistry
from superbench.benchmarks import Framework, BenchmarkRegistry
from superbench.common.utils import logger
if __name__ == '__main__':
context = BenchmarkRegistry.create_benchmark_context(
'matmul',
parameters='--num_steps=20',
'matmul', parameters='--num_steps=20', framework=Framework.PYTORCH
)
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