benchmark
356fd5da - Improve the `run_sweep.py` script to support precision attribute (#844)

Commit
3 years ago
Improve the `run_sweep.py` script to support precision attribute (#844) Summary: Rename the `fp16` option to `precision`, providing 3 options: `fp32`, `fp16`, `amp`. It also adds the `precision` attribute to the result of `run_sweep.py`. Renaming the option to `precision` makes it easier to support other precisions in the future, such as `bfloat16`. Pull Request resolved: https://github.com/pytorch/benchmark/pull/844 Reviewed By: frank-wei Differential Revision: D35296731 Pulled By: xuzhao9 fbshipit-source-id: 109dfa2a1f3becaf7a0b4105d91c36fedfe128be
Author
Parents
Loading