1// Automatically generated header file for caffe2 macros. These 2// macros are used to build the Caffe2 binary, and if you are 3// building a dependent library, they will need to be set as well 4// for your program to link correctly. 5 6#pragma once 7 8#cmakedefine CAFFE2_BUILD_SHARED_LIBS 9#cmakedefine CAFFE2_FORCE_FALLBACK_CUDA_MPI 10#cmakedefine CAFFE2_HAS_MKL_DNN 11#cmakedefine CAFFE2_HAS_MKL_SGEMM_PACK 12#cmakedefine CAFFE2_PERF_WITH_AVX 13#cmakedefine CAFFE2_PERF_WITH_AVX2 14#cmakedefine CAFFE2_THREADPOOL_MAIN_IMBALANCE 15#cmakedefine CAFFE2_THREADPOOL_STATS 16#cmakedefine CAFFE2_USE_EXCEPTION_PTR 17#cmakedefine CAFFE2_USE_ACCELERATE 18#cmakedefine CAFFE2_USE_CUDNN 19#cmakedefine CAFFE2_USE_EIGEN_FOR_BLAS 20#cmakedefine CAFFE2_USE_FBCODE 21#cmakedefine CAFFE2_USE_GOOGLE_GLOG 22#cmakedefine CAFFE2_USE_LITE_PROTO 23#cmakedefine CAFFE2_USE_MKL 24#cmakedefine USE_MKLDNN 25#cmakedefine CAFFE2_USE_NVTX 26#cmakedefine CAFFE2_USE_ITT 27 28#ifndef EIGEN_MPL2_ONLY 29#cmakedefine EIGEN_MPL2_ONLY 30#endif 31 32// Useful build settings that are recorded in the compiled binary 33// torch.__build__.show() 34#define CAFFE2_BUILD_STRINGS { \ 35 {"TORCH_VERSION", "${TORCH_VERSION}"}, \ 36 {"CXX_COMPILER", "${CMAKE_CXX_COMPILER}"}, \ 37 {"CXX_FLAGS", "${CMAKE_CXX_FLAGS}"}, \ 38 {"BUILD_TYPE", "${CMAKE_BUILD_TYPE}"}, \ 39 {"BLAS_INFO", "${BLAS_INFO}"}, \ 40 {"LAPACK_INFO", "${LAPACK_INFO}"}, \ 41 {"USE_CUDA", "${USE_CUDA}"}, \ 42 {"USE_ROCM", "${USE_ROCM}"}, \ 43 {"CUDA_VERSION", "${CUDA_VERSION}"}, \ 44 {"ROCM_VERSION", "${ROCM_VERSION}"}, \ 45 {"USE_CUDNN", "${USE_CUDNN}"}, \ 46 {"CUDNN_VERSION", "${CUDNN_VERSION}"}, \ 47 {"USE_NCCL", "${USE_NCCL}"}, \ 48 {"USE_MPI", "${USE_MPI}"}, \ 49 {"USE_GFLAGS", "${USE_GFLAGS}"}, \ 50 {"USE_GLOG", "${USE_GLOG}"}, \ 51 {"USE_GLOO", "${USE_GLOO}"}, \ 52 {"USE_NNPACK", "${USE_NNPACK}"}, \ 53 {"USE_OPENMP", "${USE_OPENMP}"}, \ 54 {"FORCE_FALLBACK_CUDA_MPI", "${CAFFE2_FORCE_FALLBACK_CUDA_MPI}"}, \ 55 {"HAS_MKL_DNN", "${CAFFE2_HAS_MKL_DNN}"}, \ 56 {"HAS_MKL_SGEMM_PACK", "${CAFFE2_HAS_MKL_SGEMM_PACK}"}, \ 57 {"PERF_WITH_AVX", "${CAFFE2_PERF_WITH_AVX}"}, \ 58 {"PERF_WITH_AVX2", "${CAFFE2_PERF_WITH_AVX2}"}, \ 59 {"USE_EXCEPTION_PTR", "${CAFFE2_USE_EXCEPTION_PTR}"}, \ 60 {"USE_ACCELERATE", "${CAFFE2_USE_ACCELERATE}"}, \ 61 {"USE_EIGEN_FOR_BLAS", "${CAFFE2_USE_EIGEN_FOR_BLAS}"}, \ 62 {"USE_LITE_PROTO", "${CAFFE2_USE_LITE_PROTO}"}, \ 63 {"USE_MKL", "${CAFFE2_USE_MKL}"}, \ 64 {"USE_MKLDNN", "${USE_MKLDNN}"}, \ 65 {"USE_NVTX", "${CAFFE2_USE_NVTX}"}, \ 66 {"USE_ITT", "${CAFFE2_USE_ITT}"}, \ 67 {"USE_ROCM_KERNEL_ASSERT", "${USE_ROCM_KERNEL_ASSERT}"}, \ 68 {"USE_CUSPARSELT", "${USE_CUSPARSELT}"}, \ 69} 70