Skip to content

Commit

Permalink
[pipelining] Add manual pipeline stage (#126123)
Browse files Browse the repository at this point in the history
Add `ManualPipelineStage` under `_PipelineStage.py`

Fix some type hints since `args_recv_info` can contain more than one RecvInfo. Previously the hint was `Tuple[InputInfo]` which meant it is a tuple of size 1. This is different from `List[InputInfo]` which can contain any number of items. I needed to update to `Tuple[InputInfo, ...]` to make the number of items flexible.

Pull Request resolved: #126123
Approved by: https://github.com/kwen2501
  • Loading branch information
H-Huang authored and ZelboK committed May 19, 2024
1 parent 1dfe2d1 commit ed27236
Showing 1 changed file with 403 additions and 6 deletions.

0 comments on commit ed27236

Please sign in to comment.