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
Vivek Khandelwal e90ea3d7ab [MLIR][TORCH] Extend implementation of aten._index_put_impl op. 
This commits adds the support for cases for index_put_op:
1.) where index is a 2-d tensor.
2.) where indices is a list of tensors and none, with exactly
2 non none tensors along the consecutive dimensions.

This commit also adds a utility to compute the broadcast shape
given the two input tensors.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 		2023-04-05 14:04:30 +05:30
..
AbstractInterpLibrary.cpp [MLIR][TORCH] Add e2e support aten.movedim.int op 2023-04-04 17:53:27 +05:30
AdjustCallingConventions.cpp update llvm to d23516e9ad477527a9db4d06b1fa9566680ac67c (#1812) 2023-01-23 16:34:22 -08:00
CMakeLists.txt Add a new RecomposeComplexOps pass, fold slice+copy_ into indeX_put_ (#1901) 2023-03-10 16:42:11 -05:00
DecomposeComplexOps.cpp [MLIR][TORCH] Extend implementation of aten._index_put_impl op. 2023-04-05 14:04:30 +05:30
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 [MLIR][TORCH] Add e2e support aten.movedim.int op 2023-04-04 17:53:27 +05:30
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 Add `RecomposeComplexOps` declaration + fix typos in pass name (#1950) 2023-03-28 11:07:47 -07:00
PrepareForGlobalizeObjectGraph.cpp update llvm to d23516e9ad477527a9db4d06b1fa9566680ac67c (#1812) 2023-01-23 16:34:22 -08:00
RecomposeComplexOps.cpp Add `RecomposeComplexOps` declaration + fix typos in pass name (#1950) 2023-03-28 11:07:47 -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
RefineTypes.cpp [MLIR][TORCH] Add decomposition for prims::squeeze op 2023-04-01 21:45:58 +05:30
ReifyAbstractInterpCalculationsUtils.cpp Cast `number` to `float` when shape function takes Scalar arg (#1978) 2023-03-28 09:30:31 -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 [custom op] Generalize shape library logic to work with dtypes (#1594) 2022-12-13 08:25:41 -08:00
SimplifyAbstractInterpCalculationsUtils.h [custom op] Generalize shape library logic to work with dtypes (#1594) 2022-12-13 08:25:41 -08:00
SimplifyDtypeCalculations.cpp Make `getTypeForScalarType` safer by returning `FailureOr<Type>` (#1814) 2023-01-20 18:40:13 +00:00
SimplifyShapeCalculations.cpp build: update llvm tag to 9acc2f37 (#1828) 2023-01-25 01:29:42 +00:00