Sylvain Jeaugey
28 satır
882 B
C
28 satır
882 B
C
/*************************************************************************
|
|
* Copyright (c) 2015, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENCE.txt for license information
|
|
************************************************************************/
|
|
|
|
|
|
// Dynamically handle dependencies on external libraries (other than cudart).
|
|
|
|
#ifndef SRC_LIBWRAP_H_
|
|
#define SRC_LIBWRAP_H_
|
|
|
|
#include "core.h"
|
|
|
|
typedef struct nvmlDevice_st* nvmlDevice_t;
|
|
|
|
ncclResult_t wrapSymbols(void);
|
|
|
|
ncclResult_t wrapNvmlInit(void);
|
|
ncclResult_t wrapNvmlShutdown(void);
|
|
ncclResult_t wrapNvmlDeviceGetHandleByPciBusId(const char* pciBusId, nvmlDevice_t* device);
|
|
ncclResult_t wrapNvmlDeviceGetIndex(nvmlDevice_t device, unsigned* index);
|
|
ncclResult_t wrapNvmlDeviceSetCpuAffinity(nvmlDevice_t device);
|
|
ncclResult_t wrapNvmlDeviceClearCpuAffinity(nvmlDevice_t device);
|
|
|
|
#endif // End include guard
|
|
|