File: macros.h.in

package info (click to toggle)
pytorch 1.7.1-7
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 80,340 kB
  • sloc: cpp: 670,830; python: 343,991; ansic: 67,845; asm: 5,503; sh: 2,924; java: 2,888; xml: 266; makefile: 244; ruby: 148; yacc: 144; objc: 51; lex: 44
file content (82 lines) | stat: -rw-r--r-- 2,954 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
// 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}"}, \
}