benchmark
39a78931 - Update torchbench commit pin, add sam_fast benchmark (#121420)

Commit
2 years ago
Update torchbench commit pin, add sam_fast benchmark (#121420) Summary: After this, the sam_fast benchmark can now be run in the pytorch repo: ``` SEGMENT_ANYTHING_FAST_USE_FLASH_4=0 benchmarks/dynamo/torchbench.py --inference --amp --performance --backend=inductor --explain --only sam_fast ``` sam_fast is designed for inference only, with cuda and amp on. The code adds these restrictions to the benchmark. X-link: https://github.com/pytorch/pytorch/pull/121420 Approved by: https://github.com/oulgen, https://github.com/msaroufim Reviewed By: osalpekar Differential Revision: D54778529 Pulled By: jamesjwu fbshipit-source-id: 63247735007cfb70ca69d67807816ee1c833170b
Author
Parents
Loading