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/IR
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
..
CMakeLists.txt Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
OpInterfaces.h Add a number of kernels and new patterns. 2020-11-04 14:36:59 -08:00
OpInterfaces.td Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
TorchBase.td Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
TorchDialect.h Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
TorchOps.h Add support for prim::Loop op. 2021-03-02 16:01:34 -08:00
TorchOps.td Add support for `prim::unchecked_cast`. 2021-03-02 16:01:34 -08:00
TorchTypes.h Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
TorchTypes.td Add support for prim.NumToTensor 2021-02-26 10:16:56 -08:00