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
History
Sean Silva df4c5764da Add support for `prim::unchecked_cast`. 
This arises when casting optionals, which happens a lot especially
around handling of default arguments (python `if arg is None` idiom).

In this case, the offending code for the model is in max_pool2d:
[code link](b3bf08e67f/torch/nn/functional.py (L657))
 		2021-03-02 16:01:34 -08:00
..
builder Add support for `prim::unchecked_cast`. 2021-03-02 16:01:34 -08:00
CMakeLists.txt Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
init_python_bindings.cpp Make torch_mlir compatible with binary PyTorch installations. 2020-12-14 09:51:00 -08:00
init_python_bindings.h Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
pybind.h Add TorchScript graph importer. 2020-11-23 14:20:09 -08:00