[NFC] polish colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp code style

ae7c3381 · zhengzangw · binmakeswell · a3b66f6d · ae7c3381
Commit ae7c3381 authored May 20, 2022 by zhengzangw Committed by binmakeswell May 20, 2022
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp +4 -4

No files found.
--- a/colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp
+++ b/colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp
@@ -30,8 +30,8 @@ void multi_tensor_lamb_cuda(int chunk_size, at::Tensor noop_flag,
                            const float max_grad_norm,
                            at::optional<bool> use_nvlamb_python);

-std::tuple<at::Tensor, at::Tensor>
-multi_tensor_l2norm_cuda(int chunk_size, at::Tensor noop_flag,
+std::tuple<at::Tensor, at::Tensor> multi_tensor_l2norm_cuda(
+    int chunk_size, at::Tensor noop_flag,
    std::vector<std::vector<at::Tensor>> tensor_lists,
    at::optional<bool> per_tensor_python);