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