cerebras.modelzoo.common.utils.run.cli_parser.add_gpu_arguments#

cerebras.modelzoo.common.utils.run.cli_parser.add_gpu_arguments(gpu_parser: argparse.ArgumentParser, first_parse: bool = True)[source]#

Injects GPU specific parser arguments.

Parameters

gpu_parser – Parser into which the arguments are being added.

previous

cerebras.modelzoo.common.utils.run.cli_parser.add_general_arguments

next

cerebras.modelzoo.common.utils.run.cli_parser.assemble_disallowlist