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以后的版本

秒客网

openmp 在android上crash的解决方案

相关文章