benchmark
644a9ceb - Rename the `enable_fp16_half()` function to `enable_fp16` (#1900)

Commit
2 years ago
Rename the `enable_fp16_half()` function to `enable_fp16` (#1900) Summary: We are renaming `enable_fp16_half()` to `enable_fp16()` because the amp fp16 is now named `enable_amp_fp16()`. This makes the naming of dargs-related functions cleaner. It also implements the enable_fp16, enable_bf16, and enable_channels_last in a cleaner by utilizing the common `inputs_cast` function. Pull Request resolved: https://github.com/pytorch/benchmark/pull/1900 Reviewed By: davidberard98 Differential Revision: D49418225 Pulled By: xuzhao9 fbshipit-source-id: 3e5e3c3af06dbc6e06a6a4a2dabcbf775dc45cec
Author
Parents
Loading