llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test/Dialect/Torch
History
Sean Silva 1e357ae680 Add simple type refinement pass. 
Currently implemented as a simple intraprocedural dataflow analysis over
a standard ShapedType lattice (hasRank, sizes, and elementType).

It currently hardcodes a few key pieces of information:
- shape transfer functions
- whether it is legal to update the operand type of an op

This needs to be made pluggable obviously and the core propagation logic
moved somewhere agnostic.
 		2021-04-07 11:06:34 -07:00
..
adjust-calling-conventions.mlir Add torch-adjust-calling-conventions pass. 2021-04-05 17:56:35 -07:00
globalize-object-graph-error.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
globalize-object-graph-free-functions.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
globalize-object-graph-initializers.mlir Give torch.global_slot an initializer region. 2021-02-26 16:24:19 -08:00
globalize-object-graph-methods.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
globalize-object-graph-module-uses-error.mlir Extend GlobalizeObjectGraph to handle torch.prim.GetAttr returning NnModuleType 2021-02-19 10:23:25 -08:00
globalize-object-graph-module-uses.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
globalize-object-graph-multiple-instances-error.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
globalize-object-graph-multiple-instances-multiple-module-args.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
globalize-object-graph-multiple-instances.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
globalize-object-graph-submodules.mlir Give torch.global_slot an initializer region. 2021-02-26 16:24:19 -08:00
globalize-object-graph-visibility.mlir Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
globalize-object-graph.mlir Give torch.global_slot an initializer region. 2021-02-26 16:24:19 -08:00
invalid.mlir Basic infra for annotate shapes and dtypes on arguments. 2021-04-01 18:40:03 -07:00
ops.mlir Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00
prepare-for-globalize-object-graph.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
refine-types.mlir Add simple type refinement pass. 2021-04-07 11:06:34 -07:00