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()