mirror of https://github.com/llvm/torch-mlir
5708ee7ec9
- Added linalg lowering for `AtenFloorDivideScalarOp` - Needed `AtenDivScalarModeOp` for the decomp. - Added linalg lowering for `AtenDivScalarModeOp` - Moved linalg payload logic to `createDivModePayload()` since the logic was nearly identical for both `AtenDivScalarModeOp` and `AtenDivTensorModeOp`. Just a template function - Added `AtenDivScalarModeOp` lowering for stablehlo Pytorch's [`torch.floor_divide()`](https://pytorch.org/docs/stable/generated/torch.floor_divide.html) in a previous version (for a reason unknown to me) preformed a truncation instead of "floor". The already implemented op `AtenFloorDivideTensorOp` was done before this change. However, this wasn't caught because our testcases only tested positive floor division. I changed this to floor as well as adding a few test cases. |
||
|---|---|---|
| .. | ||
| AbstractInterpLibrary.cpp | ||
| AdjustCallingConventions.cpp | ||
| CMakeLists.txt | ||
| DecomposeComplexOps.cpp | ||
| DropAbstractInterpCalculations.cpp | ||
| EraseModuleInitializer.cpp | ||
| FuseQuantizedOps.cpp | ||
| GlobalizeObjectGraph.cpp | ||
| InlineGlobalSlots.cpp | ||
| LowerToBackendContract.cpp | ||
| MatchQuantizedOps.cpp | ||
| MaximizeValueSemantics.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| PrepareForGlobalizeObjectGraph.cpp | ||
| RecomposeComplexOps.cpp | ||
| ReduceOpVariants.cpp | ||
| RefinePublicReturn.cpp | ||
| ReifyAbstractInterpCalculationsUtils.cpp | ||
| ReifyAbstractInterpCalculationsUtils.h | ||
| ReifyDtypeCalculations.cpp | ||
| ReifyShapeCalculations.cpp | ||
| ScalarizeShapes.cpp | ||
| SimplifyAbstractInterpCalculationsUtils.cpp | ||
| SimplifyAbstractInterpCalculationsUtils.h | ||
| SimplifyDtypeCalculations.cpp | ||
| SimplifyShapeCalculations.cpp | ||