update

Signed-off-by: limengxuan <mengxuan.li@dynamia.ai>

Author: archlitchi

src/allocator/allocator.c

@@ -195,7 +195,7 @@ int remove_chunk_only(CUdeviceptr dptr) {
             LIST_REMOVE(a_list, val);
             CUdevice dev;
             cuCtxGetDevice(&dev);
-            rm_gpu_device_memory_usage(getpid(), dev,t_size, 2);
+            rm_gpu_device_memory_usage(getpid(), dev, t_size, 2);
             return 0;
         }
     }
@@ -258,14 +258,14 @@ int add_chunk_async(CUdeviceptr *address, size_t size, CUstream hStream) {
     allocated_list_entry *e;
     INIT_ALLOCATED_LIST_ENTRY(e,addr,size);
     res = CUDA_OVERRIDE_CALL(cuda_library_entry,cuMemAllocAsync,&e->entry->address,size,hStream);
-    if (res!=CUDA_SUCCESS){
+    if (res!=CUDA_SUCCESS) {
         LOG_ERROR("cuMemoryAllocate failed res=%d",res);
         return res;
     }
     *address = e->entry->address;
     CUmemoryPool pool;
     res = CUDA_OVERRIDE_CALL(cuda_library_entry,cuDeviceGetMemPool,&pool,dev);
-    if (res != CUDA_SUCCESS){
+    if (res != CUDA_SUCCESS) {
         LOG_ERROR("cuDeviceGetMemPool failed res=%d",res);
         return res;
     }

src/cuda/memory.c

@@ -126,7 +126,7 @@ CUresult cuArrayDestroy(CUarray arr) {
     return res;
 }
 
-CUresult cuMemoryAllocate(CUdeviceptr* dptr, size_t bytesize, void* data){
+CUresult cuMemoryAllocate(CUdeviceptr* dptr, size_t bytesize, void* data) {
     CUresult res;
     res = CUDA_OVERRIDE_CALL(cuda_library_entry,cuMemAlloc_v2,dptr,bytesize);
     return res;
@@ -320,8 +320,8 @@ CUresult cuIpcCloseMemHandle(CUdeviceptr dptr){
     return CUDA_OVERRIDE_CALL(cuda_library_entry,cuIpcCloseMemHandle,dptr);
 }
 
-CUresult cuIpcGetMemHandle ( CUipcMemHandle* pHandle, CUdeviceptr dptr ){
-    LOG_MSG("cuIpcGetMemHandle dptr=%llx",dptr);
+CUresult cuIpcGetMemHandle(CUipcMemHandle* pHandle, CUdeviceptr dptr) {
+    LOG_MSG("cuIpcGetMemHandle dptr=%llx", dptr);
     ENSURE_RUNNING();
     return CUDA_OVERRIDE_CALL(cuda_library_entry,cuIpcGetMemHandle,pHandle,dptr);
 }
@@ -497,12 +497,12 @@ CUresult cuMemGetInfo_v2(size_t* free, size_t* total) {
     size_t limit = get_current_device_memory_limit(dev);
     if (limit == 0) {
         CUDA_OVERRIDE_CALL(cuda_library_entry,cuMemGetInfo_v2, free, total);
-        LOG_INFO("orig free=%ld total=%ld",*free,*total);
+        LOG_INFO("orig free=%ld total=%ld", *free, *total);
         *free = *total - usage;
-        LOG_INFO("after free=%ld total=%ld",*free,*total); 
+        LOG_INFO("after free=%ld total=%ld", *free, *total); 
         return CUDA_SUCCESS;
     } else if (limit < usage) {
-        LOG_WARN("limit < usage; usage=%ld, limit=%ld",usage,limit);
+        LOG_WARN("limit < usage; usage=%ld, limit=%ld", usage, limit);
         return CUDA_ERROR_INVALID_VALUE;
     } else {
         CUDA_OVERRIDE_CALL(cuda_library_entry,cuMemGetInfo_v2, free, total);