[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h code style (#939)

3 years ago · 44b6f8947b
parent 872aa413c2
commit 44b6f8947b
1 changed files with 4 additions and 3 deletions
--- a/colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h
+++ b/colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h
@ -20,7 +20,8 @@ void check_gpu_error(T result, char const *const func, const char *const file,
 template <typename T>
 void print_vec(const T *outv, std::string outn, int num_output_ele);

-template <typename T> T *cuda_malloc(size_t ele_num);
+template <typename T>
+T *cuda_malloc(size_t ele_num);

 void cuda_free(void *pdata);