You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
3.2 KiB
78 lines
3.2 KiB
// Copyright 2017 The Gemmlowp Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// pthread_everywhere.h: Either includes <pthread.h> or implements a
|
|
// subset of pthread functionality on top of C++11 <thread> for portability.
|
|
|
|
#ifndef GEMMLOWP_PROFILING_PTHREAD_EVERYWHERE_H_
|
|
#define GEMMLOWP_PROFILING_PTHREAD_EVERYWHERE_H_
|
|
|
|
#ifndef _WIN32
|
|
#define GEMMLOWP_USE_PTHREAD
|
|
#endif
|
|
|
|
#if defined GEMMLOWP_USE_PTHREAD
|
|
#include <pthread.h>
|
|
#else
|
|
// Implement a small subset of pthread on top of C++11 threads.
|
|
// The function signatures differ from true pthread functions in two ways:
|
|
// - True pthread functions return int error codes, ours return void.
|
|
// Rationale: the c++11 <thread> equivalent functions return void
|
|
// and use exceptions to report errors; we don't want to deal with
|
|
// exceptions in this code, so we couldn't meaningfully return errors
|
|
// in the polyfill. Also, the gemmlowp code using these pthread functions
|
|
// never checks their return values anyway.
|
|
// - True pthread *_create/*_init functions take pointers to 'attribute'
|
|
// structs; ours take nullptr_t. That is because gemmlowp always passes
|
|
// nullptr at the moment, so any support we would code for non-null
|
|
// attribs would be unused.
|
|
#include <condition_variable>
|
|
#include <cstddef>
|
|
#include <mutex>
|
|
#include <thread>
|
|
namespace gemmlowp {
|
|
using pthread_t = std::thread *;
|
|
using pthread_mutex_t = std::mutex *;
|
|
using pthread_cond_t = std::condition_variable *;
|
|
inline void pthread_create(pthread_t *thread, std::nullptr_t,
|
|
void *(*start_routine)(void *), void *arg) {
|
|
*thread = new std::thread(start_routine, arg);
|
|
}
|
|
inline void pthread_join(pthread_t thread, std::nullptr_t) { thread->join(); }
|
|
inline void pthread_mutex_init(pthread_mutex_t *mutex, std::nullptr_t) {
|
|
*mutex = new std::mutex;
|
|
}
|
|
inline void pthread_mutex_lock(pthread_mutex_t *mutex) { (*mutex)->lock(); }
|
|
inline void pthread_mutex_unlock(pthread_mutex_t *mutex) { (*mutex)->unlock(); }
|
|
inline void pthread_mutex_destroy(pthread_mutex_t *mutex) { delete *mutex; }
|
|
inline void pthread_cond_init(pthread_cond_t *cond, std::nullptr_t) {
|
|
*cond = new std::condition_variable;
|
|
}
|
|
inline void pthread_cond_signal(pthread_cond_t *cond) { (*cond)->notify_one(); }
|
|
inline void pthread_cond_broadcast(pthread_cond_t *cond) {
|
|
(*cond)->notify_all();
|
|
}
|
|
inline void pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex) {
|
|
std::unique_lock<std::mutex> lock(**mutex, std::adopt_lock);
|
|
(*cond)->wait(lock);
|
|
// detach lock from mutex so when we leave this conext
|
|
// the lock is not released
|
|
lock.release();
|
|
}
|
|
inline void pthread_cond_destroy(pthread_cond_t *cond) { delete *cond; }
|
|
} // end namespace gemmlowp
|
|
#endif
|
|
|
|
#endif // GEMMLOWP_PROFILING_PTHREAD_EVERYWHERE_H_
|