update

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

Author: archlitchi

src/allocator/allocator.c

@@ -191,8 +191,8 @@ int remove_chunk_only(CUdeviceptr dptr) {
     allocated_list_entry *val;
     for (val = a_list->head; val != NULL; val = val->next) {
         if (val->entry->address == dptr) {
-            t_size=val->entry->length;
-            LIST_REMOVE(a_list,val);
+            t_size = val->entry->length;
+            LIST_REMOVE(a_list, val);
             CUdevice dev;
             cuCtxGetDevice(&dev);
             rm_gpu_device_memory_usage(getpid(), dev,t_size, 2);
@@ -217,7 +217,8 @@ int free_raw(CUdeviceptr dptr) {
     return tmp;
 }
 
-int remove_chunk_async(allocated_list *a_list, CUdeviceptr dptr, CUstream hStream) {
+int remove_chunk_async(
+    allocated_list *a_list, CUdeviceptr dptr, CUstream hStream) {
     size_t t_size;
     if (a_list->length == 0) {
         return -1;

src/cuda/memory.c

@@ -506,12 +506,14 @@ CUresult cuMemGetInfo_v2(size_t* free, size_t* total) {
         return CUDA_ERROR_INVALID_VALUE;
     } else {
         CUDA_OVERRIDE_CALL(cuda_library_entry,cuMemGetInfo_v2, free, total);
-        LOG_INFO("orig free=%ld total=%ld limit=%ld usage=%ld",*free,*total,limit,usage);
+        LOG_INFO("orig free=%ld total=%ld limit=%ld usage=%ld",
+            *free, *total, limit,usage);
         // Ensure total memory does not exceed the physical or imposed limit.
         size_t actual_limit = (limit > *total) ? *total : limit;
         *free = (actual_limit > usage) ? (actual_limit - usage) : 0;
         *total = actual_limit;
-        LOG_INFO("after free=%ld total=%ld limit=%ld usage=%ld",*free,*total,limit,usage);
+        LOG_INFO("after free=%ld total=%ld limit=%ld usage=%ld",
+            *free, *total, limit,usage);
         return CUDA_SUCCESS;
     }
 }
@@ -602,9 +604,11 @@ CUresult cuMemMap( CUdeviceptr ptr, size_t size, size_t offset, CUmemGenericAllo
     //return 0;
 }
 
-CUresult cuMemImportFromShareableHandle ( CUmemGenericAllocationHandle* handle, void* osHandle, CUmemAllocationHandleType shHandleType ) {
+CUresult cuMemImportFromShareableHandle(
+    CUmemGenericAllocationHandle* handle, void* osHandle, CUmemAllocationHandleType shHandleType) {
     LOG_INFO("cuMemImportFromSharableHandle");
-    CUresult res = CUDA_OVERRIDE_CALL(cuda_library_entry,cuMemImportFromShareableHandle,handle,osHandle,shHandleType); 
+    CUresult res = CUDA_OVERRIDE_CALL(cuda_library_entry,
+        cuMemImportFromShareableHandle,handle,osHandle,shHandleType); 
     return res;
 }