mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
0c32e1b43e
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15363 Didn't define C10_MOBILE in the numa file move diff: D13380559 move CAFFE2_MOBILE/ANDROID/IOS to c10 ``` codemod -m -d caffe2 --extensions h,hpp,cc,cpp,mm "CAFFE2_MOBILE" "C10_MOBILE" codemod -m -d caffe2 --extensions h,hpp,cc,cpp,mm "CAFFE2_ANDROID" "C10_ANDROID" codemod -m -d caffe2 --extensions h,hpp,cc,cpp,mm "CAFFE2_IOS" "C10_IOS" ``` i-am-not-moving-c2-to-c10 Reviewed By: marcinkwiatkowski Differential Revision: D13490020 fbshipit-source-id: c4f01cacbefc0f16d5de94155c26c92fd5d780e4
84 lines
2.9 KiB
C
84 lines
2.9 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
|
|
|
|
// Caffe2 version. The plan is to increment the minor version every other week
|
|
// as a track point for bugs, until we find a proper versioning cycle.
|
|
|
|
#define CAFFE2_VERSION_MAJOR @CAFFE2_VERSION_MAJOR@
|
|
#define CAFFE2_VERSION_MINOR @CAFFE2_VERSION_MINOR@
|
|
#define CAFFE2_VERSION_PATCH @CAFFE2_VERSION_PATCH@
|
|
|
|
static_assert(
|
|
CAFFE2_VERSION_MINOR < 100,
|
|
"Programming error: you set a minor version that is too big.");
|
|
static_assert(
|
|
CAFFE2_VERSION_PATCH < 100,
|
|
"Programming error: you set a patch version that is too big.");
|
|
|
|
#define CAFFE2_VERSION \
|
|
(CAFFE2_VERSION_MAJOR * 10000 + CAFFE2_VERSION_MINOR * 100 + \
|
|
CAFFE2_VERSION_PATCH)
|
|
|
|
#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_PERF_WITH_AVX512
|
|
#cmakedefine CAFFE2_THREADPOOL_MAIN_IMBALANCE
|
|
#cmakedefine CAFFE2_THREADPOOL_STATS
|
|
#cmakedefine CAFFE2_USE_EXCEPTION_PTR
|
|
#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 CAFFE2_USE_MKLDNN
|
|
#cmakedefine CAFFE2_USE_NVTX
|
|
#cmakedefine CAFFE2_USE_TRT
|
|
|
|
#ifndef USE_NUMPY
|
|
#cmakedefine USE_NUMPY
|
|
#endif
|
|
|
|
#ifndef EIGEN_MPL2_ONLY
|
|
#cmakedefine EIGEN_MPL2_ONLY
|
|
#endif
|
|
|
|
// Useful build settings that are recorded in the compiled binary
|
|
#define CAFFE2_BUILD_STRINGS { \
|
|
{"CXX_FLAGS", "${CMAKE_CXX_FLAGS}"}, \
|
|
{"BUILD_TYPE", "${CMAKE_BUILD_TYPE}"}, \
|
|
{"BLAS", "${BLAS}"}, \
|
|
{"USE_CUDA", "${USE_CUDA}"}, \
|
|
{"USE_NCCL", "${USE_NCCL}"}, \
|
|
{"USE_MPI", "${USE_MPI}"}, \
|
|
{"USE_GFLAGS", "${USE_GFLAGS}"}, \
|
|
{"USE_GLOG", "${USE_GLOG}"}, \
|
|
{"USE_GLOO", "${USE_GLOI}"}, \
|
|
{"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}"}, \
|
|
{"PERF_WITH_AVX512", "${CAFFE2_PERF_WITH_AVX512}"}, \
|
|
{"USE_EXCEPTION_PTR", "${CAFFE2_USE_EXCEPTION_PTR}"}, \
|
|
{"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", "${CAFFE2_USE_MKLDNN}"}, \
|
|
{"USE_NVTX", "${CAFFE2_USE_NVTX}"}, \
|
|
{"USE_TRT", "${CAFFE2_USE_TRT}"}, \
|
|
{"DISABLE_NUMA", "${CAFFE2_DISABLE_NUMA}"}, \
|
|
}
|