torch-mlir

Author	SHA1	Message	Date
Ramiro Leal-Cavazos	400fecc1e5	[LINALG] Fix shape function of index.Tensor + support N-rank inputs (#972 ) This commit fixes the shape function for `index.Tensor`, adding support for multiple index tensors and `None`s in the indices list. This commit also adds support for input tensors of rank greater than 1. The lowering for `index.Tensor` still has the the limitation that only a single index tensor along the first dimension of the input tensor is supported.	2022-06-24 09:45:44 -07:00
Ramiro Leal-Cavazos	246c2df65a	[LINALG] Fix typo in conversion pattern of `aten.embedding` (#942 )	2022-06-15 09:45:10 -07:00
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