Skip to content

Commit

Permalink
Add Half support for softmax and log_softmax on CPU (pytorch#103315)
Browse files Browse the repository at this point in the history 
Add Half support for softmax and log_softmax on CPU.
Note: This introduces a correctness issue with MPS pytorch#111416 and pytorch#111479.

Pull Request resolved: pytorch#103315
Approved by: https://github.com/jgong5, https://github.com/mikaylagawarecki, https://github.com/malfet
  • Loading branch information
@CaoE @pytorchmergebot
CaoE authored and pytorchmergebot committed Oct 26, 2023
1 parent fbff99f commit 1c89ea7
Show file tree
Hide file tree
Showing 6 changed files with 185 additions and 171 deletions.
Loading

0 comments on commit 1c89ea7

Please sign in to comment.