torch-mlir

Commit Graph

Author	SHA1	Message	Date
JianzheXiao	17d02811d5	[Torch Dialect] add folder for aten.any.bool (#2388 ) * update * update * update * update * update * update * update	2023-08-30 17:29:03 +08:00
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
Ramana Radhakrishnan	738f4fe96a	Rename TorchToStd pass as TorchToArith (#1163 ) All the converters in this pass appear to create ops from the arith dialect. Hence the full rename. Fix GH Issue #409.	2022-08-10 20:12:51 +01:00

Author

SHA1

Message

Date

JianzheXiao

17d02811d5

[Torch Dialect] add folder for aten.any.bool (#2388 )

* update

* update

* update

* update

* update

* update

* update

2023-08-30 17:29:03 +08:00

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

Ramana Radhakrishnan

738f4fe96a

Rename TorchToStd pass as TorchToArith (#1163 )

All the converters in this pass appear to create ops from the
arith dialect. Hence the full rename.

Fix GH Issue #409.

2022-08-10 20:12:51 +01:00

3 Commits (eed144bfbc4cf9eafebba1949aa81f615a865eea)