benchmark
9644652f - Add torchscript jit.trace to benchmark option (#161223)

Commit
118 days ago
Add torchscript jit.trace to benchmark option (#161223) Summary: For comparing NativeRT and TorchScript. We add `torchscript-jit-trace` as an option in the benchmark. With this option, we can run trace a model and run inference with the traced module using TorchScript interpreter ``` python ./benchmarks/dynamo/huggingface.py --performance --inference --torchscript-jit-trace python ./benchmarks/dynamo/timm_models.py --performance --inference --torchscript-jit-trace python ./benchmarks/dynamo/torchbench.py --performance --inference --torchscript-jit-trace ``` X-link: https://github.com/pytorch/pytorch/pull/161223 Approved by: https://github.com/huydhn Reviewed By: seemethere, henryoier Differential Revision: D80911933 fbshipit-source-id: 5c81dec32fc63f8fa2d7f4c5f71b919d31eb74b4
Author
Parents
Loading