infiniccl_cuda.h 417 Bytes
Newer Older
1
2
3
4
5
6
#ifndef INFINICCL_CUDA_H_
#define INFINICCL_CUDA_H_

#include "../infiniccl_impl.h"

// Windows does not support CUDA
wooway777's avatar
wooway777 committed
7
#if (defined(ENABLE_NVIDIA_API) || defined(ENABLE_ILUVATAR_API) || defined(ENABLE_QY_API) || defined(ENABLE_HYGON_API) || defined(ENABLE_ALI_API)) && defined(ENABLE_CCL) && !defined(_WIN32)
8
9
10
11
12
13
INFINICCL_DEVICE_API_IMPL(cuda)
#else
INFINICCL_DEVICE_API_NOOP(cuda)
#endif

#endif /* INFINICCL_CUDA_H_ */