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. |
||
|---|---|---|
| .. | ||
| _torch_mlir_custom_op_example | ||
| csrc | ||
| jit_ir_importer | ||
| _dynamo_fx_importer.py | ||
| _version.py | ||
| compiler_utils.py | ||
| dynamo.py | ||
| torchscript.py | ||