openmp 在android上crash的解决方案
这是个GOMP已知的问题,参见 bug42616, bug52738。如果在非主线程上使用openmp指令或者函数,
会crash。这是因为在android上gomp_thread(libgomp/libgomp.h文件中)函数对于用户创建的线程返回NULL
#ifdef HAVE_TLS
extern __thread struct gomp_thread gomp_tls_data;
static inline struct gomp_thread *gomp_thread (void)
{
return &gomp_tls_data;
}
#else
extern pthread_key_t gomp_tls_key;
static inline struct gomp_thread *gomp_thread (void)
{
return pthread_getspecific (gomp_tls_key);
}
#endif
参见上附代码,GOMP 在有无tls时的实现不同:
- 如果有tls,HAVE_TLS defined,使用全局变量跟踪每个线程状态
- 反之,线程局部数据通过pthread_setspecific管理
在ndk的早期版本中不支持tls,__thread也没有,所以HAVE_TLS undefined,所以使用了pthread_setspecific函数
当GOMP创建一个工作线程时,它在gomp_thread_start()(line 72)设置了线程specific数据:
#ifdef HAVE_TLS
thr = &gomp_tls_data;
#else
struct gomp_thread local_thr;
thr = &local_thr;
pthread_setspecific (gomp_tls_key, thr);
#endif
但是,当应用创建一个非主thread时,线程specific数据没有被设置,所以gomp_thread()函数返回NULL。
这就导致了crash,但是如果支持了tls就不会有这个问题
解决方案有两个:
- 在每个线程创建时调用以下代码:
// call this piece of code from App Main UI thread at beginning of the App execution,
// and then again in JNI call that will invoke the OpenMP calculations
void initTLS_2_fix_bug_42616_and_52738() {
extern pthread_key_t gomp_tls_key;
static void * _p_gomp_tls = NULL;
void *ptr = pthread_getspecific(gomp_tls_key);
if (ptr == NULL) {
pthread_setspecific(gomp_tls_key, _p_gomp_tls);
} else {
_p_gomp_tls = ptr;
}
}
- 这篇文章中提供了两个patch,需要重新编译gcc
- 使用ndk-10e以后的版本