llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,016 Commits
47 Branches
1195 Tags
96 MiB
Commit Graph

4 Commits (07d92bcbb4439fa5eb524090f4a8a6bd121ac473)

Author SHA1 Message Date
Vivek Khandelwal 4b11284440 [MLIR][TORCH] Add E2E support for aten.avg_pool2d op 
This commit adds lowering of `aten.avg_pool2d` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2022-05-02 12:31:44 +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
Vivek Khandelwal 769f3a8870 [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 
This commit adds lowering of `max_pool2d_with_indices` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2022-04-18 21:05:19 +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