#pragma once
|
|
#ifndef C10_USING_CUSTOM_GENERATED_MACROS
|
#include <c10/cuda/impl/cuda_cmake_macros.h>
|
#endif
|
|
// See c10/macros/Export.h for a detailed explanation of what the function
|
// of these macros are. We need one set of macros for every separate library
|
// we build.
|
|
#ifdef _WIN32
|
#if defined(C10_CUDA_BUILD_SHARED_LIBS)
|
#define C10_CUDA_EXPORT __declspec(dllexport)
|
#define C10_CUDA_IMPORT __declspec(dllimport)
|
#else
|
#define C10_CUDA_EXPORT
|
#define C10_CUDA_IMPORT
|
#endif
|
#else // _WIN32
|
#if defined(__GNUC__)
|
#define C10_CUDA_EXPORT __attribute__((__visibility__("default")))
|
#else // defined(__GNUC__)
|
#define C10_CUDA_EXPORT
|
#endif // defined(__GNUC__)
|
#define C10_CUDA_IMPORT C10_CUDA_EXPORT
|
#endif // _WIN32
|
|
// This one is being used by libc10_cuda.so
|
#ifdef C10_CUDA_BUILD_MAIN_LIB
|
#define C10_CUDA_API C10_CUDA_EXPORT
|
#else
|
#define C10_CUDA_API C10_CUDA_IMPORT
|
#endif
|
|
/**
|
* The maximum number of GPUs that we recognizes.
|
*/
|
#define C10_COMPILE_TIME_MAX_GPUS 16
|
