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