experimental_experiment.torch_bench.dort_bench_profile¶

Profiles an execution¶

The datas should be saved with script dort_bench.py and option --export <something>.

python -m experimental_experiment.torch_bench.dort_bench_profile --help

Example, run llama model with onnxrt backend on cuda.

python -m experimental_experiment.torch_bench.dort_bench_profile \
       --model model.onnx --inputs model.onnx.pkl

experimental_experiment.torch_bench.dort_bench_profile.main()[source]¶: Main function for command line python -m experimental_experiment.torch_bench.dort_bench_profile.