torch-mlir

Commit Graph

Author	SHA1	Message	Date
Alexandre Rames	224ee27610	Fix a few missing dependencies. (#2014 ) `TorchToTMTensor` depends on `TorchMLIRTorchUtils` for `mlir::torch::torch_upstream::get_reduction_enum`. `TorchMLIRTorchConversionPasses` depends on multiple libs for both tblgen'd headers and definitions. Test with `ninja TorchMLIRTorchConversionPasses` from a clean build.	2023-04-11 11:18:49 -07:00
Maksim Levental	829717c96e	Bump LLVM (#958 )	2022-06-22 22:23:46 -05:00
Vivek Khandelwal	1a2a9e066f	[MLIR][TORCH] Add TorchToTMTensor pass This pass is added to lower ops, which can not be lowered via the TorchToLinalg pass, such as `torch.bincount` op. This pass also uses torch-mlir's TMTensor Dialect to lower the complex ops. Also add torch.bincount op lowering with the help of TMTensor dialect Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>	2022-03-08 22:52:34 +05:30

Author

SHA1

Message

Date

Alexandre Rames

224ee27610

Fix a few missing dependencies. (#2014 )

`TorchToTMTensor` depends on `TorchMLIRTorchUtils` for
`mlir::torch::torch_upstream::get_reduction_enum`.

`TorchMLIRTorchConversionPasses` depends on multiple libs for both tblgen'd
headers and definitions. Test with `ninja TorchMLIRTorchConversionPasses` from
a clean build.

2023-04-11 11:18:49 -07:00

Maksim Levental

829717c96e

Bump LLVM (#958 )

2022-06-22 22:23:46 -05:00

Vivek Khandelwal

1a2a9e066f

[MLIR][TORCH] Add TorchToTMTensor pass

This pass is added to lower ops, which can not be lowered
via the TorchToLinalg pass, such as `torch.bincount` op.
This pass also uses torch-mlir's TMTensor Dialect to lower the
complex ops.

Also add torch.bincount op lowering with the help of TMTensor dialect

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>

2022-03-08 22:52:34 +05:30

3 Commits (5698893ae424ffaf30923a38256b8bd4a1dcd344)