fix: 修复参数传递问题

2026-03-30 22:22:36 +08:00
parent e1f9901384
commit 345fd2f091
6 changed files with 13 additions and 5 deletions
@@ -81,6 +81,7 @@ def only_on_rank(rank, sync=False):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
+            ret_args = None
            if get_rank() == rank:
                ret_args = func(*args, **kwargs)