Toggle Light / Dark / Auto color theme
Toggle table of contents sidebar
Source code for experimental_experiment.torch_bench.bash_bench_issues
"""
Benchmark exporters
===================
Benchmarks many models coming from issues.
Available exporters:
* eager: identity
* export: :func:`torch.export.export`
* compile: :func:`torch.compile`
* custom: :func:`experimental_experiment.torch_interpreter.to_onnx`
* torch_script: :func:`torch.onnx.export`
* onnx_dynamo: :func:`torch.onnx.export` with ``dynamo=True``
* dynamo_export: :func:`torch.onnx.dynamo_export`
::
python -m experimental_experiment.torch_bench.bash_bench_issues --help
::
python -m experimental_experiment.torch_bench.bash_bench_issues --model ""
::
python -m experimental_experiment.torch_bench.bash_bench_issues \\
--model FluxT5 --exporter eager
"""
from experimental_experiment.torch_bench._bash_bench_cmd import bash_bench_main
[docs]
def main ( args = None ):
"""
Main function for command line
``python -m experimental_experiment.torch_bench.bash_bench_issues``.
"""
bash_bench_main ( "bash_bench_issues" , __doc__ , args )
if __name__ == "__main__" :
main ()