hotfix colotensor _scan_for_pg_from_args (#1276)

7aadcbd0 · ver217 · GitHub · 0cf8e8e9 · 7aadcbd0
Unverified Commit 7aadcbd0 authored Jul 12, 2022 by ver217 Committed by GitHub Jul 12, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

colossalai/tensor/colo_tensor.py colossalai/tensor/colo_tensor.py +2 -2

No files found.
--- 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