mirror of https://github.com/llvm/torch-mlir
140babd952
A recent PyTorch commit made ConstantPad2d call a helper function with a `Union[int, float]` type annotated. This commit adds minimal support for representing and dealing with that. https://github.com/pytorch/pytorch/pull/73287 Changes: - Adding support for `!torch.union<T1, T2, T3>`/`Torch::UnionType`, along with the importer and CAPI code. - Add support in isValidSubtype for union types. - Adding a canonicalizer for `torch.derefine` to help simplify some code that derefines to a UnionType (this also fixes #664). There is still more work to do for really supporting UnionType well, such as canonicalizing UnionType's so that they can be compared with pointer equality. |
||
|---|---|---|
| .. | ||
| GlobalizeObjectGraph | ||
| adjust-calling-conventions.mlir | ||
| canonicalize.mlir | ||
| decompose-complex-ops.mlir | ||
| drop-shape-calculations.mlir | ||
| inline-global-slots.mlir | ||
| invalid.mlir | ||
| maximize-value-semantics.mlir | ||
| ops.mlir | ||
| prepare-for-globalize-object-graph.mlir | ||
| promote-types.mlir | ||
| reduce-op-variants-error.mlir | ||
| reduce-op-variants.mlir | ||
| refine-public-return.mlir | ||
| refine-types-branch.mlir | ||
| refine-types-ops.mlir | ||
| refine-types.mlir | ||
| reify-shape-calculations.mlir | ||
| simplify-shape-calculations.mlir | ||