From 944e2441c612c5727742eb4ad7eab52483e8dc9f Mon Sep 17 00:00:00 2001 From: Longjie Zheng Date: Mon, 2 Sep 2024 15:46:15 +0200 Subject: [PATCH] nit --- optimum/fx/parallelization/op_registry/op_handlers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/fx/parallelization/op_registry/op_handlers.py b/optimum/fx/parallelization/op_registry/op_handlers.py index 18042c3e2b9..56b8fc16bc0 100644 --- a/optimum/fx/parallelization/op_registry/op_handlers.py +++ b/optimum/fx/parallelization/op_registry/op_handlers.py @@ -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: