github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-07 04:18:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
yuxuan-lou 2022-05-13 15:37:03 +08:00 committed by binmakeswell
parent 872aa413c2
commit 44b6f8947b
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h
View File

@ -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);