torch-mlir/lib/Dialect/Torch/IR
Sean Silva 8860b5c55d Add `torch.prim.If`
This removes the use of `scf.if`, which required laundering back and
forth between `i1` and `!torch.bool` in the frontend. We will eventually
lower this op to `scf.if`, but this results in a cleaner IR and layering
at the frontend.
2021-06-16 14:04:31 -07:00
..
CMakeLists.txt Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
TorchDialect.cpp Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
TorchOps.cpp Add `torch.prim.If` 2021-06-16 14:04:31 -07:00
TorchTypes.cpp Add `!torch.tuple<T1, T2>` type. 2021-06-15 08:15:22 -07:00