torch-mlir

Commit Graph

Author	SHA1	Message	Date
Vivek Khandelwal	c69a1e5688	[MLIR][TORCH] Add E2E support for ScalarImplicit, Int.Scalar op This commit adds lowering of `aten.ScalarImplicit` and `aten.Int.Scalar` op. Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>	2022-05-10 22:40:49 +05:30
Prashant Kumar	5cdef0213d	[LINALG] Bug fix i64 vs i32 type comparison. Comparing index type instead of integer types solves the problem.	2022-04-22 08:09:58 +05:30
Sean Silva	5d9222383c	Split up TorchToLinalg.cpp This helps keep things organized and also exposes more parallelism to the build system. It seems though that most of the compile time is actually spent in the headers though, so the wall time doesn't decrease as much as I had hoped (and now that the headers are being included multiple times, the cpu time actually increases a lot, sadly -- will try to dig into this).	2022-03-14 10:19:41 -07:00

Author

SHA1

Message

Date

Vivek Khandelwal

c69a1e5688

[MLIR][TORCH] Add E2E support for ScalarImplicit, Int.Scalar op

This commit adds lowering of `aten.ScalarImplicit` and `aten.Int.Scalar` op.

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

2022-05-10 22:40:49 +05:30

Prashant Kumar

5cdef0213d

[LINALG] Bug fix i64 vs i32 type comparison.

Comparing index type instead of integer types solves the problem.

2022-04-22 08:09:58 +05:30

Sean Silva

5d9222383c

Split up TorchToLinalg.cpp

This helps keep things organized and also exposes more parallelism to
the build system. It seems though that most of the compile time is
actually spent in the headers though, so the wall time doesn't decrease
as much as I had hoped (and now that the headers are being included
multiple times, the cpu time actually increases a lot, sadly -- will try
to dig into this).

2022-03-14 10:19:41 -07:00

3 Commits (cfc1a6515c00badbf15bff8e3f2e796d3199b8e0)