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/test
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
..
acap_export Add support for prim::GetAttr/SetAttr/CallMethod/If 2021-02-04 17:01:47 -08:00
builder Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
ivalue_import Properly import the entire torch::jit::CompilationUnit 2021-03-01 12:08:01 -08:00
node_import Add support for `prim::unchecked_cast`. 2021-03-02 16:01:34 -08:00
CMakeLists.txt Enable building using LLVM_EXTERNAL_PROJECTS. (#152) 2021-01-26 11:43:43 -07:00
extension_coexistence.py Update test configuration to import mlir from LLVM install location. 2020-10-12 15:25:07 -07:00
lit.cfg.py Enable building using LLVM_EXTERNAL_PROJECTS. (#152) 2021-01-26 11:43:43 -07:00
lit.site.cfg.py.in Enable building using LLVM_EXTERNAL_PROJECTS. (#152) 2021-01-26 11:43:43 -07:00