| 12345678910111213141516171819202122 |
- #pragma once
- // Test these using #if AT_MKL_ENABLED(), not #ifdef, so that it's
- // obvious if you forgot to include Config.h
- // c.f. https://stackoverflow.com/questions/33759787/generating-an-error-if-checked-boolean-macro-is-not-defined
- //
- // DO NOT put the macros for CUDA libraries in this file; they belong in cuda/CUDAConfig.h
- #define AT_MKLDNN_ENABLED() 0
- #define AT_MKL_ENABLED() 0
- #define AT_MKL_SEQUENTIAL() 0
- #define AT_FFTW_ENABLED() 0
- #define AT_POCKETFFT_ENABLED() 1
- #define AT_NNPACK_ENABLED() 1
- #define CAFFE2_STATIC_LINK_CUDA() 0
- #define AT_BUILD_WITH_BLAS() 1
- #define AT_BUILD_WITH_LAPACK() 1
- #define AT_PARALLEL_OPENMP 1
- #define AT_PARALLEL_NATIVE 0
- #define AT_PARALLEL_NATIVE_TBB 0
- #define AT_BLAS_F2C() 0
- #define AT_BLAS_USE_CBLAS_DOT() 1
|
