torch-mlir

Author SHA1 Message Date

Author	SHA1	Message	Date
Ramiro Leal-Cavazos	00a6e9c1bb	[LINALG] Add value tensor variant to `fill_.Scalar` (#600 ) This commit adds the op `PseudoAtenFillScalarOp` that represents `AtenFill_ScalarOp` without the underscore. The approach is the same as in commit `dd998fa4d4`. Adding this op allows for a simpler and more consistent version of the `empty` and `empty_like` op e2e tests.	2022-02-15 11:58:03 -08:00
Ramiro Leal-Cavazos	c1167853db	Fix error in RefineTypes for constant alloc ops (#579 ) This commit fixes an error in the refine types pass of constant allocation ops. The function used to set the dtype, `fillInDtypeGivenDtypeAndDataType`, takes two torch types as arguments, but a torch type and a standard MLIR type were being passed into it. This commit also fixes the way the dtype was calculated in `visitAtenToDtypeOp`. This op was also passing a standard MLIR type as an argument to the `fillInDtypeGivenDtypeAndDataType` function. Moreover, since the op `aten.to.dtype` has the dtype argument as not optional, all that is needed is to match against the int value to extract the dtype.	2022-02-10 18:02:18 -08:00
Gaurav Shukla	3c40539b34	[TORCH][MLIR] Add E2E support for `aten.[ones_like\|zeros_like]` - This commit adds E2E support for `aten.ones_like` and `aten.zeros_like` ops. - Adds support for non-None `dtype` argument of `aten.empty_like` op. - All the unit test cases related to constant tensor allocation like ops are moved to a different file named `constant_alloc.py`. Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>	2022-01-06 20:24:40 +05:30

Ramiro Leal-Cavazos

[LINALG] Add value tensor variant to `fill_.Scalar` (#600 )

This commit adds the op `PseudoAtenFillScalarOp` that represents
`AtenFill_ScalarOp` without the underscore. The approach is the same
as in commit dd998fa4d4.

Adding this op allows for a simpler and more consistent version of the
`empty` and `empty_like` op e2e tests.

2022-02-15 11:58:03 -08:00

Ramiro Leal-Cavazos

c1167853db

Fix error in RefineTypes for constant alloc ops (#579 )

This commit fixes an error in the refine types pass of constant
allocation ops. The function used to set the dtype,
`fillInDtypeGivenDtypeAndDataType`, takes two torch types as arguments,
but a torch type and a standard MLIR type were being passed into it.

This commit also fixes the way the dtype was calculated in
`visitAtenToDtypeOp`. This op was also passing a standard MLIR type as
an argument to the `fillInDtypeGivenDtypeAndDataType`
function. Moreover, since the op `aten.to.dtype` has the dtype
argument as not optional, all that is needed is to match
against the int value to extract the dtype.

2022-02-10 18:02:18 -08:00

Gaurav Shukla

3c40539b34

[TORCH][MLIR] Add E2E support for `aten.[ones_like|zeros_like]`

- This commit adds E2E support for `aten.ones_like` and
  `aten.zeros_like` ops.
- Adds support for non-None `dtype` argument of `aten.empty_like` op.
- All the unit test cases related to constant tensor allocation like ops
  are moved to a different file named `constant_alloc.py`.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

2022-01-06 20:24:40 +05:30

3 Commits (9b2613533bd78632b4438fa0b11319cd9d86f9e0)