1 | #pragma once |
2 | |
3 | // Test these using #if AT_MKL_ENABLED(), not #ifdef, so that it's |
4 | // obvious if you forgot to include Config.h |
5 | // c.f. https://stackoverflow.com/questions/33759787/generating-an-error-if-checked-boolean-macro-is-not-defined |
6 | // |
7 | // DO NOT put the macros for CUDA libraries in this file; they belong in cuda/CUDAConfig.h |
8 | |
9 | #define AT_MKLDNN_ENABLED() 1 |
10 | #define AT_MKL_ENABLED() 0 |
11 | #define AT_MKL_SEQUENTIAL() 0 |
12 | #define AT_FFTW_ENABLED() 0 |
13 | #define AT_POCKETFFT_ENABLED() 1 |
14 | #define AT_NNPACK_ENABLED() 1 |
15 | #define CAFFE2_STATIC_LINK_CUDA() 0 |
16 | #define AT_BUILD_WITH_BLAS() 0 |
17 | #define AT_BUILD_WITH_LAPACK() 0 |
18 | #define AT_PARALLEL_OPENMP 1 |
19 | #define AT_PARALLEL_NATIVE 0 |
20 | #define AT_PARALLEL_NATIVE_TBB 0 |
21 | #define AT_BLAS_F2C() 0 |
22 | #define AT_BLAS_USE_CBLAS_DOT() 0 |
23 | |