Make Mish support large inputs. (#43037)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43037
In the previous version of mish_op.cc, the output would be 'nan' for large inputs. We re-write mish_op.cc to solve this problem.
Test Plan:
Unit test
buck test //dper3/dper3/modules/tests:core_modules_test -- test_linear_compress_embedding_with_attention_with_activation_mish
{F284052906}
buck test mode/opt //dper3/dper3_models/ads_ranking/tests:model_paradigm_e2e_tests -- test_sparse_nn_with_mish
{F284224158}
## Workflow
f212113434
{F285281318}
Differential Revision: D23102644
fbshipit-source-id: 98f1ea82f8c8e05b655047b4520c600fc1a826f4