mirror of https://github.com/llvm/torch-mlir
de02b56e17
This commit adds dtype functions for all the torch ops that did not previously have one and removes the pass `RefineTypes`, since the abstract interpretation library now takes care of all the dtype propagation. All dtype functions added are tested except for - `aten.embedding` - `aten._embedding_bag` - `aten.embedding_bag` These functions need a change to the testing framework to allow specifying the actual data inside the tensor used for testing. I will fix this in a follow up patch. Co-authored-by: Jiahao Li <liplus17@163.com> |
||
|---|---|---|
| .. | ||
| Torch | ||
| TorchConversion | ||
| CMakeLists.txt | ||