/usr/local/lib64/python3.6/site-packages/torch/include/c10/cuda
NameSizeModeActions
impl/-0755rm
CUDACachingAllocator.h49870644editdlrm
CUDAException.h31080644editdlrm
CUDAFunctions.h28760644editdlrm
CUDAGraphsC10Utils.h29690644editdlrm
CUDAGuard.h112120644editdlrm
CUDAMacros.h9830644editdlrm
CUDAMathCompat.h36030644editdlrm
CUDAMiscFunctions.h2540644editdlrm
CUDAStream.h89930644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/c10/cuda/CUDAMacros.h (983B)
#pragma once #ifndef C10_USING_CUSTOM_GENERATED_MACROS #include #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