apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

init (#6642)

Browse Source 
Co-authored-by: gongweibao <gognweibao@baidu.com>
This commit is contained in:
gongweibao
2026-03-04 21:55:31 +08:00
committed by GitHub
parent 5c8f5184d9
commit ddb06ff83f
306 changed files with 40627 additions and 34418 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_ops/gpu_ops/cuda_multiprocess.h
+20 -19
View File
@@ -52,35 +52,36 @@ namespace cub = hipcub;
#define GPU(str) cuda##str
#endif
#define checkCudaErrors(call) \
do { \
GPU(Error_t) err = call; \
if (err != GPU(Success)) { \
printf("CUDA error at %s %d: %s\n", \
__FILE__, \
__LINE__, \
GPU(GetErrorString)(err)); \
exit(EXIT_FAILURE); \
} \
} while (0)
#define checkCudaErrors(call) \
do { \
GPU(Error_t) err = call; \
if (err != GPU(Success)) { \
printf("CUDA error at %s %d: %s\n", \
__FILE__, \
__LINE__, \
GPU(GetErrorString)(err)); \
exit(EXIT_FAILURE); \
} \
} while (0)
typedef struct shmStruct_st {
size_t nprocesses;
GPU(IpcMemHandle_t) memHandle;
size_t nprocesses;
GPU(IpcMemHandle_t) memHandle;
} shmStruct;
typedef struct sharedMemoryInfo_st {
void *addr;
size_t size;
void *addr;
size_t size;
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
HANDLE shmHandle;
HANDLE shmHandle;
#else
int shmFd;
int shmFd;
#endif
} sharedMemoryInfo;
inline int sharedMemoryOpen(const char *name, size_t sz, sharedMemoryInfo *info) {
inline int sharedMemoryOpen(const char *name,
size_t sz,
sharedMemoryInfo *info) {
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
info->size = sz;