hotfix colotensor _scan_for_pg_from_args (#1276)

2022-07-12 20:46:31 +08:00 · 2022-07-12 20:46:31 +08:00 · 7aadcbd070
parent 0cf8e8e91c
commit 7aadcbd070
1 changed files with 2 additions and 2 deletions
--- a/colossalai/tensor/colo_tensor.py
+++ b/colossalai/tensor/colo_tensor.py
@ -40,7 +40,7 @@ def _scan_for_pg_from_args(args, kwargs) -> ProcessGroup:
            pg = _scan_for_pg_from_args(elem, {})
            if pg is not None:
                return pg
-    for k, v in kwargs:
+    for k, v in kwargs.items():
        if isinstance(v, ColoTensor):
            pg = v.get_process_group()
            return pg