AMDMIGraphX
AMDMIGraphX copied to clipboard
Update `pass_manager` to handle updating the cached shape when submodule shapes change
- Extend code in
pass_manager.cpp:run_passes(prog, root_mod, passes, trace)to handle updating the cached shape fromop.compute_shape(shape_inputs, mod_inputs)whenmod_inputsshape changes.- Currently, the cached shape is not updated when submodule inputs have their output shape changed.
- Can probably be implemented by keeping track of the parent instructions that need to be updated after all submodules are updated.
- Remove the
simplify_select_module_output_shapematcher and pass fromsimplify_dyn_ops.cpp. Move the shape computation logic intoselect_module.compute_shape().