History

Bryce Arden 68338eafb7 [chore] Make variable names in prim.py more clear		2021-03-02 17:28:32 -08:00
..
README.md	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
add3.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
debug-info.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
elif.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
errors.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
if.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
list.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
loop.py	Add support for prim::Loop op.	2021-03-02 16:01:34 -08:00
prim.py	[chore] Make variable names in prim.py more clear	2021-03-02 17:28:32 -08:00
tuple.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
types-bool.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00
types-none.py	Rename tests to match the code they test	2021-02-25 13:31:33 -08:00

README.md

node_import

Most of the tests in this directory test the importing of TorchScript torch::jit::Graph's.

However, TorchScript graphs don't really correspond directly to anything on the MLIR side. They are a weird combination of a context, builder, and function and just holds a torch::jit::Block. It is torch::jit::Node and torch::jit::Block which form the recursive structure analogous to MLIR's operation/region/block.

torch::jit::Node == mlir::Operation,
torch::jit::Block == mlir::Region containing single mlir::Block

Hence the name of this directory and the corresponding code in node_importer.h/cpp.