Skip to content

Commit

Permalink
Fix PassConfig docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
ProExpertProg authored Nov 14, 2024
1 parent 234ea39 commit bbb6300
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions vllm/compilation/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ class PassConfig(BaseModel):
Each pass defines its own stages (before, after, maybe in-between).
- dump_graph_dir: directory to dump the graphs. Default is .
- enable_fusion: whether to enable the custom fusion pass.
- enable_reshape: whether to enable the custom reshape elimination pass.
TODO better pass enabling system.
"""
dump_graph_stages: List[str] = Field(default_factory=list)
Expand Down

0 comments on commit bbb6300

Please sign in to comment.