[NFC] polish colossalai/nn/_ops/embedding.py code style (#1561)

0c4c9aa6 · BigOneLiXiaoMing · Frank Lee · 08815f0e · 0c4c9aa6
Commit 0c4c9aa6 authored Sep 08, 2022 by BigOneLiXiaoMing Committed by Frank Lee Sep 08, 2022
Show whitespace changes
Inline Side-by-side

Showing with 10 additions and 11 deletions

colossalai/nn/_ops/embedding.py colossalai/nn/_ops/embedding.py +10 -11

No files found.
--- a/colossalai/nn/_ops/embedding.py
+++ b/colossalai/nn/_ops/embedding.py
@@ -113,8 +113,7 @@ def colo_embedding(input_tensor: GeneralTensor,
    if not weight.has_compute_spec():    # No Model Parallel Applied
        assert weight.is_replicate(), 'Invalid weight spec for native embedding op'
-        return ColoTensor.from_torch_tensor(
+        return ColoTensor.from_torch_tensor(tensor=F.embedding(input_tensor,
-            tensor=F.embedding(input_tensor,
                                                               weight,
                                                               padding_idx=padding_idx,
                                                               max_norm=max_norm,