Ensure that the module we consume has no unused computations. This can causes issues as we clone modules to support try_multiple_mesh_shapes, and cloning an HLO module removes dead computations leading to mismatches.
Ensure that the module we consume has no unused computations. This can causes issues as we clone modules to support try_multiple_mesh_shapes, and cloning an HLO module removes dead computations leading to mismatches.
