Sylvain Jeaugey
28189e2df8
Add support for CUDA 12.0, drop Kepler (sm_35). Support for H100 features. Make socket code more robust and protected. Solves #555. Improve performance on large CUDA graphs, reducing dependencies. Reduce inter-socket bandwidth on AMD CPUs to favor better paths. Various fixes to ncclCommAbort. Make service thread polling resistant to EINTR. Compile with profiling API by default. Extend NVTX instrumentation with call arguments.
18 regels
597 B
C
18 regels
597 B
C
/*************************************************************************
|
|
* Copyright (c) 2016-2022, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef NCCL_SHM_H_
|
|
#define NCCL_SHM_H_
|
|
|
|
#include "nccl.h"
|
|
|
|
typedef void* ncclShmHandle_t;
|
|
ncclResult_t ncclShmOpen(char* shmPath, size_t shmSize, void** shmPtr, void** devShmPtr, int refcount, ncclShmHandle_t* handle);
|
|
ncclResult_t ncclShmClose(ncclShmHandle_t handle);
|
|
ncclResult_t ncclShmUnlink(ncclShmHandle_t handle);
|
|
|
|
#endif
|