githubmirror/ColossalAI

colossalai/cli/cli.py

24 строки372 B

Frank Lee

[cli] refactored micro-benchmarking cli and added more metrics (#858)

25 апр 2022, 06:48

25 апр 2022, 06:48a82da26

import click
from .launcher import run
from .check import check
from .benchmark import benchmark


class Arguments():

    def __init__(self, arg_dict):
        for k, v in arg_dict.items():
            self.__dict__[k] = v


@click.group()
def cli():
    pass


cli.add_command(run)
cli.add_command(check)
cli.add_command(benchmark)

if __name__ == '__main__':
    cli()