torch-mlir/lib/Dialect/Torch/Transforms
Zhekun Zhang a426363b7d
[Torch Dialect] Add split.tensor support + recompose rules (#2102)
* add split.tensor support + recompose rules

* add e2e test

* address comments

* address comments

* erase op in recomposeOp

---------

Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-23 12:43:33 -07:00
..
AbstractInterpLibrary.cpp build: manually update PyTorch version 2023-05-18 21:30:11 +05:30
AdjustCallingConventions.cpp update llvm to d23516e9ad477527a9db4d06b1fa9566680ac67c (#1812) 2023-01-23 16:34:22 -08:00
CMakeLists.txt Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
DecomposeComplexOps.cpp [Torch Dialect] require dtype exists when decompose to aten.where.self (#2094) 2023-05-17 09:04:26 -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
GlobalizeObjectGraph.cpp update llvm to d23516e9ad477527a9db4d06b1fa9566680ac67c (#1812) 2023-01-23 16:34:22 -08:00
InlineGlobalSlots.cpp update llvm to d23516e9ad477527a9db4d06b1fa9566680ac67c (#1812) 2023-01-23 16:34:22 -08:00
LowerToBackendContract.cpp Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
MaximizeValueSemantics.cpp build: update llvm tag to 7ccbb4df (#1736) 2022-12-20 18:17:27 +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] Add split.tensor support + recompose rules (#2102) 2023-05-23 12:43:33 -07:00
ReduceOpVariants.cpp handles 2,3,4 from https://github.com/llvm/torch-mlir/issues/1963 (#1964) 2023-03-24 21:50:01 -05:00
RefinePublicReturn.cpp Add support for returning more than one copy of the same tensor (#1228) 2022-08-18 22:41:45 +00:00
ReifyAbstractInterpCalculationsUtils.cpp Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -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 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
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 Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
SimplifyShapeCalculations.cpp Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00