Skip to content

Commit

Permalink
nit
Browse files Browse the repository at this point in the history
  • Loading branch information
zhenglongjiepheonix committed Sep 2, 2024
1 parent c87f881 commit 944e244
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions optimum/fx/parallelization/op_registry/op_handlers.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@

class Registry:
"""
Registry class handles registration of parallel axis propagation handlers of different aten ops, to support a new
aten op, you need to register the corresponding handler class by decorating it with `register` function.
Registry class handles registration of parallel axis propagation handlers of different aten ops.
To support a new aten op, you need to register the corresponding handler class by decorating it with `register` function.
"""

def __init__(self) -> None:
Expand Down

0 comments on commit 944e244

Please sign in to comment.