mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
1433bc1455
The check is for older compilers and is now aways true. Pull Request resolved: https://github.com/pytorch/pytorch/pull/147247 Approved by: https://github.com/janeyx99
69 lines
2.4 KiB
C
69 lines
2.4 KiB
C
// Automatically generated header file for caffe2 macros. These
|
|
// macros are used to build the Caffe2 binary, and if you are
|
|
// building a dependent library, they will need to be set as well
|
|
// for your program to link correctly.
|
|
|
|
#pragma once
|
|
|
|
#cmakedefine CAFFE2_BUILD_SHARED_LIBS
|
|
#cmakedefine CAFFE2_FORCE_FALLBACK_CUDA_MPI
|
|
#cmakedefine CAFFE2_HAS_MKL_DNN
|
|
#cmakedefine CAFFE2_HAS_MKL_SGEMM_PACK
|
|
#cmakedefine CAFFE2_PERF_WITH_AVX
|
|
#cmakedefine CAFFE2_PERF_WITH_AVX2
|
|
#cmakedefine CAFFE2_THREADPOOL_MAIN_IMBALANCE
|
|
#cmakedefine CAFFE2_THREADPOOL_STATS
|
|
#cmakedefine CAFFE2_USE_ACCELERATE
|
|
#cmakedefine CAFFE2_USE_CUDNN
|
|
#cmakedefine CAFFE2_USE_EIGEN_FOR_BLAS
|
|
#cmakedefine CAFFE2_USE_FBCODE
|
|
#cmakedefine CAFFE2_USE_GOOGLE_GLOG
|
|
#cmakedefine CAFFE2_USE_LITE_PROTO
|
|
#cmakedefine CAFFE2_USE_MKL
|
|
#cmakedefine USE_MKLDNN
|
|
#cmakedefine CAFFE2_USE_NVTX
|
|
#cmakedefine CAFFE2_USE_ITT
|
|
|
|
#ifndef EIGEN_MPL2_ONLY
|
|
#cmakedefine EIGEN_MPL2_ONLY
|
|
#endif
|
|
|
|
// Useful build settings that are recorded in the compiled binary
|
|
// torch.__build__.show()
|
|
#define CAFFE2_BUILD_STRINGS { \
|
|
{"TORCH_VERSION", "${TORCH_VERSION}"}, \
|
|
{"CXX_COMPILER", "${CMAKE_CXX_COMPILER}"}, \
|
|
{"CXX_FLAGS", "${CMAKE_CXX_FLAGS}"}, \
|
|
{"BUILD_TYPE", "${CMAKE_BUILD_TYPE}"}, \
|
|
{"BLAS_INFO", "${BLAS_INFO}"}, \
|
|
{"LAPACK_INFO", "${LAPACK_INFO}"}, \
|
|
{"USE_CUDA", "${USE_CUDA}"}, \
|
|
{"USE_ROCM", "${USE_ROCM}"}, \
|
|
{"CUDA_VERSION", "${CUDA_VERSION}"}, \
|
|
{"ROCM_VERSION", "${ROCM_VERSION}"}, \
|
|
{"USE_CUDNN", "${USE_CUDNN}"}, \
|
|
{"COMMIT_SHA", "${COMMIT_SHA}"}, \
|
|
{"CUDNN_VERSION", "${CUDNN_VERSION}"}, \
|
|
{"USE_NCCL", "${USE_NCCL}"}, \
|
|
{"USE_MPI", "${USE_MPI}"}, \
|
|
{"USE_GFLAGS", "${USE_GFLAGS}"}, \
|
|
{"USE_GLOG", "${USE_GLOG}"}, \
|
|
{"USE_GLOO", "${USE_GLOO}"}, \
|
|
{"USE_NNPACK", "${USE_NNPACK}"}, \
|
|
{"USE_OPENMP", "${USE_OPENMP}"}, \
|
|
{"FORCE_FALLBACK_CUDA_MPI", "${CAFFE2_FORCE_FALLBACK_CUDA_MPI}"}, \
|
|
{"HAS_MKL_DNN", "${CAFFE2_HAS_MKL_DNN}"}, \
|
|
{"HAS_MKL_SGEMM_PACK", "${CAFFE2_HAS_MKL_SGEMM_PACK}"}, \
|
|
{"PERF_WITH_AVX", "${CAFFE2_PERF_WITH_AVX}"}, \
|
|
{"PERF_WITH_AVX2", "${CAFFE2_PERF_WITH_AVX2}"}, \
|
|
{"USE_ACCELERATE", "${CAFFE2_USE_ACCELERATE}"}, \
|
|
{"USE_EIGEN_FOR_BLAS", "${CAFFE2_USE_EIGEN_FOR_BLAS}"}, \
|
|
{"USE_LITE_PROTO", "${CAFFE2_USE_LITE_PROTO}"}, \
|
|
{"USE_MKL", "${CAFFE2_USE_MKL}"}, \
|
|
{"USE_MKLDNN", "${USE_MKLDNN}"}, \
|
|
{"USE_NVTX", "${CAFFE2_USE_NVTX}"}, \
|
|
{"USE_ITT", "${CAFFE2_USE_ITT}"}, \
|
|
{"USE_ROCM_KERNEL_ASSERT", "${USE_ROCM_KERNEL_ASSERT}"}, \
|
|
{"USE_CUSPARSELT", "${USE_CUSPARSELT}"}, \
|
|
}
|