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
zjgarvey 55ff110dc2
[MLIR][TORCH] Only unroll prim loop-like ops within a `torch.shape.calculate` region (#3812) 
Reports a match failure for the pattern `FullyUnrollPrimLoop` when the
loop op is not in a region defined by a `torch.shape.calculate` op.

This is needed to avoid unrolling prim loops generated by ONNX IR, since
we are applying shape refinement in the
`torch-onnx-to-torch-backend-pipeline` introduced in fa4794d .

See also the discussion in
<https://github.com/iree-org/iree/pull/18867#discussion_r1811101655>
 		2024-10-23 13:38:55 +05:30
..
AbstractInterpLibrary.cpp build: manually update PyTorch version (#3727) 2024-10-18 13:32:14 +05:30
AdjustCallingConventions.cpp Update to llvm/llvm-proect@27ac46e6be (2024-6-12) (#3454) 2024-06-12 19:34:01 -07:00
CMakeLists.txt Add RestructureNonConstantAxes pass to address reduce op tests failing on non constant axes (#3600) 2024-08-26 14:06:06 -07:00
DecomposeComplexOps.cpp [Torch] support adaptive_max_pool1d when return_indices equals False (#3783) 2024-10-11 23:42:15 +08:00
DropAbstractInterpCalculations.cpp Update to LLVM 029313cc979ae71877b65794b1063d4e51184cc8 2023-03-21 04:16:20 -07:00
EraseModuleInitializer.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
FuseQuantizedOps.cpp [torch] Basic support for per-channel quantized graphs (#3623) 2024-08-10 15:51:09 +02:00
GlobalizeObjectGraph.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
InlineGlobalSlots.cpp Integrate LLVM at llvm/llvm-project@c13f806 (#3789) 2024-10-14 15:00:45 +02:00
LowerToBackendContract.cpp Add Decompostion for `Aten_SafeSoftmaxOp` (#3708) 2024-09-12 16:58:10 -05:00
MatchQuantizedOps.cpp [torch] Basic support for per-channel quantized graphs (#3623) 2024-08-10 15:51:09 +02:00
MaximizeValueSemantics.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp [MLIR][TORCH] Add torch-onnx-to-torch-backend pipeline (#3801) 2024-10-21 11:20:44 -05:00
PrepareForGlobalizeObjectGraph.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
RecomposeComplexOps.cpp [Torch] Add support for Meshgrid (#3462) 2024-06-14 23:59:08 +08:00
ReduceOpVariants.cpp build: manually update PyTorch version (#3627) 2024-08-19 12:03:56 +05:30
RefinePublicReturn.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
ReifyAbstractInterpCalculationsUtils.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07: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 Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
ReifyShapeCalculations.cpp Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
RestructureNonConstantAxes.cpp Add RestructureNonConstantAxes pass to address reduce op tests failing on non constant axes (#3600) 2024-08-26 14:06:06 -07:00
ScalarizeShapes.cpp Add More Scalarize Shapes Patterns (#3810) 2024-10-21 19:42:39 -05:00
SimplifyAbstractInterpCalculationsUtils.cpp [MLIR][TORCH] Only unroll prim loop-like ops within a `torch.shape.calculate` region (#3812) 2024-10-23 13:38:55 +05:30
SimplifyAbstractInterpCalculationsUtils.h Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
SimplifyDtypeCalculations.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
SimplifyShapeCalculations.cpp Add AtenSliceTOp Canonicalization to SimplifyShapeCalculations pass (#3791) 2024-10-14 14:41:31 -05:00