search

llvm / torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
Other
1.35k stars 505 forks source link

Reference for decomposition of coarser torch ops into finer torch ops. #542

Closed pashu123 closed 2 years ago

pashu123 commented 2 years ago

https://github.com/pytorch/functorch/blob/main/functorch/_src/decompositions.py

This is a good reference for breaking/decomposing torch ops.

pashu123 commented 2 years ago

Modified the softmax decomposition here: https://github.com/llvm/torch-mlir/pull/554

pashu123 commented 2 years ago

Modified log_softmax decomposition here: https://github.com/llvm/torch-mlir/pull/580

silvasean commented 2 years ago

Closing this as it is more "documentation" rather than an actual issue. (also the decompositions are now spread out between _decomps and _refs)