torch-mlir/include/npcomp/Dialect/Torch/IR
Yi Zhang bfc3ee35c6 Import Machine Translation model to MLIR.
This includes the following changes to import MT model into MLIR. There
are still a lot of work to for actual compilation.
- Add `torch.dict<>`, `torch.any`, `torch.number` types
- Add `torch.prim.DictConstruct` op
- Fix `torch.prim.TupleConstruct` op assembly format to include resulting types
2021-08-10 15:22:06 -04:00
..
.gitignore Significantly restructure torch/aten import design. 2021-05-19 13:37:39 -07:00
CMakeLists.txt Bump llvm-project to da289a174fc6617c7be37be2947480510fd4f02a 2021-07-07 13:57:29 -07:00
GeneratedAtenOps.td Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00
GeneratedPrimOps.td Add CastOpInterface to torch.prim.unchecked_cast. 2021-06-23 08:07:45 -07:00
GeneratedQuantizedOps.td Add a !torch.float type. 2021-06-17 09:24:18 -07:00
TorchBase.td Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
TorchDialect.h Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
TorchOps.h Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d 2021-07-09 15:04:29 -07:00
TorchOps.td Import Machine Translation model to MLIR. 2021-08-10 15:22:06 -04:00
TorchTraits.h Significantly restructure torch/aten import design. 2021-05-19 13:37:39 -07:00
TorchTypes.h Move Torch type implementation code into TorchTypes.cpp 2021-06-10 16:46:47 -07:00
TorchTypes.td Import Machine Translation model to MLIR. 2021-08-10 15:22:06 -04:00