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

cerebras.modelzoo.common.utils.run.cli_parser.add_gpu_arguments(gpu_parser)[source]#

Injects GPU specific parser arguments.

Parameters

gpu_parser (argparse.ArgumentParser) – Parser into which the arguments are being added.