Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 334 Bytes
Newer Older
guoshzhao's avatar
Benchmarks: Initialization - Add base class, registry, and result (#1)  
guoshzhao committed
1
2
3
4
5 
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

"""A module containing all the micro-benchmarks."""
guoshzhao's avatar
Benchmarks: Code Revision - Move benchmarks auto-registration from registry.py to __init__.py (#24)  
guoshzhao committed
6 
from superbench.benchmarks.micro_benchmarks.micro_base import MicroBenchmark
guoshzhao's avatar
Benchmarks: Add Benchmark - Add op-sharding microbenchmark, including matmul...  
guoshzhao committed
7 
from superbench.benchmarks.micro_benchmarks.sharding_matmul import ShardingMatmul
guoshzhao's avatar
Benchmarks: Initialization - Add base class, registry, and result (#1)  
guoshzhao committed
8
guoshzhao's avatar
Benchmarks: Add Benchmark - Add op-sharding microbenchmark, including matmul...  
guoshzhao committed
9 
__all__ = ['MicroBenchmark', 'ShardingMatmul']