llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/include/npcomp/Dialect/Torch
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
..
IR Add support for `prim::unchecked_cast`. 2021-03-02 16:01:34 -08:00
Transforms Give torch.global_slot an initializer region. 2021-02-26 16:24:19 -08:00
CMakeLists.txt Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00