experimental_experiment.torch_bench._bash_bench_cmd

experimental_experiment.torch_bench._bash_bench_cmd.bash_bench_main(script_name: str, doc: str, args: List[str] | None = None)[source]

Main command line for all bash_bench script.

Parameters:
  • script_name – suffix for the bash

  • doc – documentation

  • args – optional arguments

experimental_experiment.torch_bench._bash_bench_cmd.bash_bench_parse_args(name: str, doc: str, new_args: List[str] | None = None)[source]

Returns parsed arguments.