torch-mlir/lib/Dialect/Torch/Transforms
zjgarvey 40e762ca42
Adds result types to a prelu decomp (#3098)
This adds explicit result types instead of relying on shape/dtype
computations.

Solves a regression issue with IREE: #3092
2024-04-02 11:41:56 -07:00
..
AbstractInterpLibrary.cpp Implement linalg lowering of diag_embed torch op (#2885) 2024-03-22 16:32:50 -07:00
AdjustCallingConventions.cpp Bump LLVM to llvm/llvm-project@5fcf907b34 (#2810) 2024-01-26 18:38:44 -08:00
CMakeLists.txt [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -07:00
DecomposeComplexOps.cpp Adds result types to a prelu decomp (#3098) 2024-04-02 11:41:56 -07:00
DropAbstractInterpCalculations.cpp Update to LLVM 029313cc979ae71877b65794b1063d4e51184cc8 2023-03-21 04:16:20 -07:00
EraseModuleInitializer.cpp update llvm to d23516e9ad477527a9db4d06b1fa9566680ac67c (#1812) 2023-01-23 16:34:22 -08:00
FuseQuantizedOps.cpp [ONNX] Preliminary Work Towards Supporting QuantizedMLP_basic onnx e2e test (#3089) 2024-04-01 16:21:05 -07:00
GlobalizeObjectGraph.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
InlineGlobalSlots.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
LowerToBackendContract.cpp [Torch] Support prelu decomposition (#3069) 2024-03-29 08:05:00 +08:00
MatchQuantizedOps.cpp [torch][quant] Quantized `torch.mm` for linalg with end-to-end test (#2750) 2024-01-24 14:02:50 -08:00
MaximizeValueSemantics.cpp Bump LLVM to llvm/llvm-project@5fcf907b34 (#2810) 2024-01-26 18:38:44 -08:00
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
PrepareForGlobalizeObjectGraph.cpp update llvm to d23516e9ad477527a9db4d06b1fa9566680ac67c (#1812) 2023-01-23 16:34:22 -08:00
RecomposeComplexOps.cpp [Torch Dialect] support aten.split_with_sizes (#2431) 2023-09-04 09:59:26 +08:00
ReduceOpVariants.cpp [torch] Fix tm_tensor.attention for end-to-end (#2907) 2024-02-13 21:18:01 -08:00
RefinePublicReturn.cpp Support `DerefineOp` in `RefinePublicReturn`. 2023-07-20 20:08:46 +02:00
ReifyAbstractInterpCalculationsUtils.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
ReifyAbstractInterpCalculationsUtils.h handles 2,3,4 from https://github.com/llvm/torch-mlir/issues/1963 (#1964) 2023-03-24 21:50:01 -05:00
ReifyDtypeCalculations.cpp handles 2,3,4 from https://github.com/llvm/torch-mlir/issues/1963 (#1964) 2023-03-24 21:50:01 -05:00
ReifyShapeCalculations.cpp handles 2,3,4 from https://github.com/llvm/torch-mlir/issues/1963 (#1964) 2023-03-24 21:50:01 -05:00
ScalarizeShapes.cpp [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -07:00
SimplifyAbstractInterpCalculationsUtils.cpp Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
SimplifyAbstractInterpCalculationsUtils.h Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
SimplifyDtypeCalculations.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
SimplifyShapeCalculations.cpp Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00