| #ifndef CAFFE2_UTILS_THREADPOOL_COMMON_H_ | |
| #define CAFFE2_UTILS_THREADPOOL_COMMON_H_ | |
| #ifdef __APPLE__ | |
| #include <TargetConditionals.h> | |
| #endif | |
| // caffe2 depends upon NNPACK, which depends upon this threadpool, so | |
| // unfortunately we can't reference core/common.h here | |
| // This is copied from core/common.h's definition of C10_MOBILE | |
| // Define enabled when building for iOS or Android devices | |
| #if defined(__ANDROID__) | |
| #define C10_ANDROID 1 | |
| #elif (defined(__APPLE__) && \ | |
| (TARGET_IPHONE_SIMULATOR || TARGET_OS_SIMULATOR || TARGET_OS_IPHONE)) | |
| #define C10_IOS 1 | |
| #elif (defined(__APPLE__) && TARGET_OS_MAC) | |
| #define C10_IOS 1 | |
| #else | |
| #endif // ANDROID / IOS / MACOS | |
| #endif // CAFFE2_UTILS_THREADPOOL_COMMON_H_ |