Qiskit / qiskit / 26909282146

87%

Build Type

push

github

Committed by web-flow

Commit Message

Refactor ProgramNode::call to flatten+call_flat+unflatten (#16298)

* Refactor ProgramNode to flat call_flat + DataTree flatten/unflatten helpers

Split the ProgramNode trait so implementations only define
`call_flat(args: &[Tensor]) -> Result<Vec<Tensor>, _>` while a blanket
ProgramNodeExt provides the DataTree-shaped `call(args: &DataTree<Tensor>)`
on top. Adds the supporting DataTree machinery (`flatten_against`,
`unflatten`, `map_leaves`, `into_leaves`, `TreeMatchError`,
`ArityMismatch`) and structured call errors (`CallInputError`, `CallError`,
`MissingCallError`). Store is rewritten on top of `call_flat` and stores
its tensors as a flat `Vec<Tensor>` aligned with `output_types`.

* Fix double iteration in unflatten()

* three small changes from code review

* inlined helper functions into the functions they help

* write a custom iterator struct for into_leaves()

* add docstring for IntoLeaves::stack

Coverage Stats

220 of 250 new or added lines in 3 files covered. (88.0%)

25 existing lines in 3 files now uncovered.

108913 of 124513 relevant lines covered (87.47%)

961235.69 hits per line