benchmark
9df52151 - Add special device mapping (#1752)

Commit
2 years ago
Add special device mapping (#1752) Summary: To reuse the inference batch size already set in models' meta.yaml file for different kind of devices like AMD GPUs, this PR add a `SPECIAL_DEVICE_MAPPING` and got the corresponding batch size when executes `determine_batch_size ()` Pull Request resolved: https://github.com/pytorch/benchmark/pull/1752 Reviewed By: xuzhao9 Differential Revision: D47277610 Pulled By: FindHao fbshipit-source-id: 188df62eb5089b4729559c288e1ed7401984a2a0
Author
FindHao
Parents
Loading