Browse Source

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

pull/1012/head
zhengzangw 3 years ago committed by binmakeswell
parent
commit
ae7c338105
  1. 8
      colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp

8
colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp

@ -30,10 +30,10 @@ 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::vector<std::vector<at::Tensor>> tensor_lists,
at::optional<bool> per_tensor_python);
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);
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("multi_tensor_scale", &multi_tensor_scale_cuda,

Loading…
Cancel
Save