github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(cuda): encapsulate asynchronous allocation methods.

Browse Source
This commit is contained in:
Pedro Alves
2022-11-07 12:37:48 -03:00
committed by Agnès Leroy
parent 13e77b2d8c
commit cf222e9176
3 changed files with 34 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/device.h
View File

@@ -7,6 +7,8 @@ int cuda_destroy_stream(void *v_stream, uint32_t gpu_index);
void *cuda_malloc(uint64_t size, uint32_t gpu_index);
void *cuda_malloc_async(uint64_t size, void *v_stream);
int cuda_check_valid_malloc(uint64_t size, uint32_t gpu_index);
int cuda_memcpy_to_cpu(void *dest, const void *src, uint64_t size,
@@ -26,5 +28,7 @@ int cuda_synchronize_device(uint32_t gpu_index);
int cuda_drop(void *ptr, uint32_t gpu_index);
int cuda_drop_async(void *ptr, void *v_stream);
int cuda_get_max_shared_memory(uint32_t gpu_index);
}