tree b7e1ae664779e675cd1edaaad86de27b6d53febf
parent 97eba8e1745e582a8faa246ec325a30c40506873
author Ryo <ryo.suzuki@arm.com> 1722534888 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1722534888 +0000

Enable oneDNN for tanh based GELU on aarch64 (#130925)

Provides speedup for GELU on aarch64 compared to native PyTorch implementation. e.g.

  8.5x speedup compared to native implementation for 1x1x16384 on 32 threads on Graviton 3

Pull Request resolved: https://github.com/pytorch/pytorch/pull/130925
Approved by: https://github.com/malfet