52 lines
1.4 KiB
C
52 lines
1.4 KiB
C
#pragma once
|
|
|
|
#ifndef C10_USING_CUSTOM_GENERATED_MACROS
|
|
|
|
// We have not yet modified the AMD HIP build to generate this file so
|
|
// we add an extra option to specifically ignore it.
|
|
#ifndef C10_CUDA_NO_CMAKE_CONFIGURE_FILE
|
|
#include <c10/cuda/impl/cuda_cmake_macros.h>
|
|
#endif // C10_CUDA_NO_CMAKE_CONFIGURE_FILE
|
|
|
|
#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. Increasing this beyond the
|
|
* initial limit of 16 broke Caffe2 testing, hence the ifdef guards.
|
|
* This value cannot be more than 128 because our DeviceIndex is a uint8_t.
|
|
o */
|
|
#ifdef FBCODE_CAFFE2
|
|
// fbcode depends on this value being 16
|
|
#define C10_COMPILE_TIME_MAX_GPUS 16
|
|
#else
|
|
#define C10_COMPILE_TIME_MAX_GPUS 120
|
|
#endif
|