21 lines
677 B
C
21 lines
677 B
C
#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
|
|
#endif // ANDROID / IOS
|
|
|
|
#endif // CAFFE2_UTILS_THREADPOOL_COMMON_H_
|