Skip to content

Commit

Permalink
onnx.MelWeightMatrix Onnx to Torch to Linalg (llvm#3659)
Browse files Browse the repository at this point in the history 
- This PR adds new (and equivalent) more tensorized impl of
MelWeightMatrix which lowers all the way to linalg.
- [Ref Pytorch
Impl](https://gist.github.com/PhaneeshB/4e6dfcded3007b1b686fbe28f07a67cd)
- Thanks to @rsuderman for pointing out the difficulties [earlier
impl](llvm#3503) posed during lowering to linalg and also for providing a
better numpy impl 🙏
  • Loading branch information
@PhaneeshB
PhaneeshB authored Aug 22, 2024
1 parent fcc5f44 commit 9a6fe58
Show file tree
Hide file tree
Showing 2 changed files with 330 additions and 319 deletions.
Loading

0 comments on commit 9a6fe58

Please sign in to comment.