You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
770 B
23 lines
770 B
#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() 1
|
|
#define AT_MKLDNN_ACL_ENABLED() 0
|
|
#define AT_MKL_ENABLED() 1
|
|
#define AT_MKL_SEQUENTIAL() 0
|
|
#define AT_POCKETFFT_ENABLED() 0
|
|
#define AT_NNPACK_ENABLED() 0
|
|
#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() 0
|