llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch/Transforms
History
Xinyu Yang da88efad89
[Torch] Fix bug of DecomposeAtenSelectIntOp (#3087) 
Fix bug of DecomposeAtenSelectIntOp. Because it may use resultTy when
resultTy has not been inferred.

```
    auto resultTy = op.getType().cast<BaseTensorType>();
    if (sliceTy.getSizes().size() == resultTy.getSizes().size()) {
      rewriter.replaceOp(op, slice);
      return success();
    }

```

So I add restriction.
 		2024-04-01 21:25:02 +08: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 [Torch] Fix bug of DecomposeAtenSelectIntOp (#3087) 2024-04-01 21:25:02 +08: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] Convert `onnx.QLinearConv` to `torch` (#2851) 2024-02-05 16:09:41 -08: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