pytorch
68034197 - [ONNX] Support gelu for fp16 export (#50487) (#50911)

Commit
3 years ago
[ONNX] Support gelu for fp16 export (#50487) (#50911) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50911 Need to replace dtype of export created scalars from float to double. (In torch implicit conversion logic, python numbers are double) Test case skipped in CI due to that current CI job env does not have CUDA support. Test Plan: Imported from OSS Reviewed By: pbelevich Differential Revision: D26050889 Pulled By: SplitInfinity fbshipit-source-id: 1fdde23a68d4793e6b9a82840acc213e5c3aa760
Author
Parents
Loading