pytorch
eae0f3f5 - Add mkl implementation for exponential on CPU (#69967)

Commit
2 years ago
Add mkl implementation for exponential on CPU (#69967) ### Description Add mkl implementation for exponential on CPU to improve the performance of exponential. ### Testing data type: float32 single socket (28cores): ``` before: torch.Size([10, 128, 10, 124]) 0.065 s torch.Size([10, 128, 20, 124]) 0.130 s after: torch.Size([10, 128, 10, 124]) 5.9e-05 s torch.Size([10, 128, 20, 124]) 0.000113 s ``` single core: ``` before: torch.Size([10, 128, 10, 124]) 0.065 s torch.Size([10, 128, 20, 124]) 0.130 s after: torch.Size([10, 128, 10, 124]) 0.00117 s torch.Size([10, 128, 20, 124]) 0.002347 s ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/69967 Approved by: https://github.com/frank-wei, https://github.com/jgong5
Author
Committer
Parents
Loading