llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/csrc/builder
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
..
acap_dispatch.cpp Fix signature of unboxed aten::arange for torch HEAD 2021-03-18 17:53:52 -07:00
acap_dispatch.h Fix signature of unboxed aten::arange for torch HEAD 2021-03-18 17:53:52 -07:00
class_annotator.cpp [cleanup] Put the root class type for exportPath first. 2021-04-01 18:40:03 -07:00
class_annotator.h [cleanup] Put the root class type for exportPath first. 2021-04-01 18:40:03 -07:00
debug.cpp Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
debug.h Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
func_builder.cpp Add support for prim::GetAttr/SetAttr/CallMethod/If 2021-02-04 17:01:47 -08:00
func_builder.h Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
function_importer.cpp Basic infra for annotate shapes and dtypes on arguments. 2021-04-01 18:40:03 -07:00
function_importer.h Basic infra for annotate shapes and dtypes on arguments. 2021-04-01 18:40:03 -07:00
ivalue_importer.cpp Miscellaneous changes while trying to work on ResNet18 2021-04-27 11:51:11 -07:00
ivalue_importer.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
mlir_utils.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
module_builder.cpp Bump llvm-project to e31c77b1827fa4dd3511f21af11cfab18ecf6d38 2021-03-10 11:01:16 -08:00
module_builder.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
node_importer.cpp Miscellaneous changes while trying to work on ResNet18 2021-04-27 11:51:11 -07:00
node_importer.h Properly model "derefinement". 2021-03-03 15:09:44 -08:00
op_builder.cpp Add prim::Print and fix prim::CallMethod 2021-02-10 15:15:56 -08:00
op_builder.h Add prim::Print and fix prim::CallMethod 2021-02-10 15:15:56 -08:00
python_bindings.cpp Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
torch_to_mlir_utils.cpp Bump llvm-project to 484b6648fdd4b104eaf7a2504dd07b60af2c9f8d 2021-04-22 18:12:55 -07:00
torch_to_mlir_utils.h Properly model "derefinement". 2021-03-03 15:09:44 -08:00