llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion
History
Sean Silva 55c3cc6624 Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 
Interestingly, TorchScript has its own op (`torch::jit::Operator`)
registry separate from the dispatcher (it is a superset of the
dispatcher).

This is where the "prim" ops and some "aten" ops (that should probably
be renamed to "prim") live. In particular, `aten::__is__` is in that
latter category of "aten but really prim". This registry is also the
source of truth for what the TorchScript interpreter calls into when it
executes.

The bulk of the "not part of the dispatcher" ops live in
09feb5f579/torch/csrc/jit/runtime/register_prim_ops.cpp (L82)

And the registry itself lives in:
09feb5f579/torch/csrc/jit/runtime/operator.cpp (L196)

This fold further reduces the IR of ResNet by folding away some
more not-taken branches. These not-taken branches in ResNet require
first-class handling of the list type which we don't yet have on any
backend.
 		2021-04-30 10:57:02 -07:00
..
ATenToLinalg Bump llvm-project to 12011b5217929ef8a56c2099c6f3233934ea4fbc 2021-04-27 13:12:33 -07:00
ATenToStd Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
ATenToTCF Add `aten.mm` to linalg lowering. 2021-04-16 12:03:31 -07:00
BasicpyToStd Bump llvm-project to 12011b5217929ef8a56c2099c6f3233934ea4fbc 2021-04-27 13:12:33 -07:00
NumpyToTCF Bump llvm-project to 0524a09cc7e1a0797982feacf505825231efbee7 2021-03-23 14:29:05 -07:00
TCFToLinalg Bump llvm-project to 12011b5217929ef8a56c2099c6f3233934ea4fbc 2021-04-27 13:12:33 -07:00
TCFToStd Bump llvm-project to 12011b5217929ef8a56c2099c6f3233934ea4fbc 2021-04-27 13:12:33 -07:00
TCFToTCP Bump llvm-project to 12011b5217929ef8a56c2099c6f3233934ea4fbc 2021-04-27 13:12:33 -07:00
CMakeLists.txt Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
PassDetail.h Initial TCF/TCP E2E seed. 2020-05-08 20:20:41 -07:00
Passes.cpp Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00