xref: /aosp_15_r20/external/pytorch/caffe2/utils/threadpool/ThreadPoolCommon.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker #ifndef CAFFE2_UTILS_THREADPOOL_COMMON_H_
2*da0073e9SAndroid Build Coastguard Worker #define CAFFE2_UTILS_THREADPOOL_COMMON_H_
3*da0073e9SAndroid Build Coastguard Worker 
4*da0073e9SAndroid Build Coastguard Worker #ifdef __APPLE__
5*da0073e9SAndroid Build Coastguard Worker #include <TargetConditionals.h>
6*da0073e9SAndroid Build Coastguard Worker #endif
7*da0073e9SAndroid Build Coastguard Worker 
8*da0073e9SAndroid Build Coastguard Worker // caffe2 depends upon NNPACK, which depends upon this threadpool, so
9*da0073e9SAndroid Build Coastguard Worker // unfortunately we can't reference core/common.h here
10*da0073e9SAndroid Build Coastguard Worker 
11*da0073e9SAndroid Build Coastguard Worker // This is copied from core/common.h's definition of C10_MOBILE
12*da0073e9SAndroid Build Coastguard Worker // Define enabled when building for iOS or Android devices
13*da0073e9SAndroid Build Coastguard Worker #if defined(__ANDROID__)
14*da0073e9SAndroid Build Coastguard Worker #define C10_ANDROID 1
15*da0073e9SAndroid Build Coastguard Worker #elif (defined(__APPLE__) &&                                            \
16*da0073e9SAndroid Build Coastguard Worker        (TARGET_IPHONE_SIMULATOR || TARGET_OS_SIMULATOR || TARGET_OS_IPHONE))
17*da0073e9SAndroid Build Coastguard Worker #define C10_IOS 1
18*da0073e9SAndroid Build Coastguard Worker #endif // ANDROID / IOS
19*da0073e9SAndroid Build Coastguard Worker 
20*da0073e9SAndroid Build Coastguard Worker #endif  // CAFFE2_UTILS_THREADPOOL_COMMON_H_
21