ncnn / build /src /platform.h

thanks to ncnn ❤

be903e2 about 3 years ago

7.25 kB

// Tencent is pleased to support the open source community by making ncnn available.
//
// Copyright (C) 2017 THL A29 Limited, a Tencent company. All rights reserved.
//
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
// in compliance with the License. You may obtain a copy of the License at
//
// https://opensource.org/licenses/BSD-3-Clause
//
// 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.

#ifndef NCNN_PLATFORM_H
#define NCNN_PLATFORM_H

#define NCNN_STDIO 1
#define NCNN_STRING 1
#define NCNN_SIMPLEOCV 0
#define NCNN_SIMPLEOMP 0
#define NCNN_SIMPLESTL 0
#define NCNN_THREADS 1
#define NCNN_BENCHMARK 0
#define NCNN_C_API 1
#define NCNN_PLATFORM_API 1
#define NCNN_PIXEL 1
#define NCNN_PIXEL_ROTATE 1
#define NCNN_PIXEL_AFFINE 1
#define NCNN_PIXEL_DRAWING 1
#define NCNN_VULKAN 0
#define NCNN_SYSTEM_GLSLANG 0
#define NCNN_RUNTIME_CPU 1
#define NCNN_GNU_INLINE_ASM 1
#define NCNN_AVX 1
#define NCNN_XOP 1
#define NCNN_FMA 1
#define NCNN_F16C 1
#define NCNN_AVX2 1
#define NCNN_AVXVNNI 1
#define NCNN_AVX512 1
#define NCNN_AVX512VNNI 1
#define NCNN_AVX512BF16 0
#define NCNN_AVX512FP16 0
#define NCNN_VFPV4 0
#define NCNN_ARM82 0
#define NCNN_ARM82DOT 0
#define NCNN_ARM82FP16FML 0
#define NCNN_ARM84BF16 0
#define NCNN_ARM84I8MM 0
#define NCNN_ARM86SVE 0
#define NCNN_ARM86SVE2 0
#define NCNN_ARM86SVEBF16 0
#define NCNN_ARM86SVEI8MM 0
#define NCNN_ARM86SVEF32MM 0
#define NCNN_MSA 0
#define NCNN_LSX 0
#define NCNN_MMI 0
#define NCNN_RVV 0
#define NCNN_INT8 1
#define NCNN_BF16 1
#define NCNN_FORCE_INLINE 1

#define NCNN_VERSION_STRING "1.0.20230913"

#include "ncnn_export.h"

#ifdef __cplusplus

#if NCNN_THREADS
#if (defined _WIN32 && !(defined __MINGW32__))
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <process.h>
#else
#include <pthread.h>
#endif
#endif // NCNN_THREADS

#if __ANDROID_API__ >= 26
#define VK_USE_PLATFORM_ANDROID_KHR
#endif // __ANDROID_API__ >= 26

namespace ncnn {

#if NCNN_THREADS
#if (defined _WIN32 && !(defined __MINGW32__))
class NCNN_EXPORT Mutex
{
public:
    Mutex() { InitializeSRWLock(&srwlock); }
    ~Mutex() {}
    void lock() { AcquireSRWLockExclusive(&srwlock); }
    void unlock() { ReleaseSRWLockExclusive(&srwlock); }
private:
    friend class ConditionVariable;
    // NOTE SRWLock is available from windows vista
    SRWLOCK srwlock;
};

class NCNN_EXPORT ConditionVariable
{
public:
    ConditionVariable() { InitializeConditionVariable(&condvar); }
    ~ConditionVariable() {}
    void wait(Mutex& mutex) { SleepConditionVariableSRW(&condvar, &mutex.srwlock, INFINITE, 0); }
    void broadcast() { WakeAllConditionVariable(&condvar); }
    void signal() { WakeConditionVariable(&condvar); }
private:
    CONDITION_VARIABLE condvar;
};

static unsigned __stdcall start_wrapper(void* args);
class NCNN_EXPORT Thread
{
public:
    Thread(void* (*start)(void*), void* args = 0) { _start = start; _args = args; handle = (HANDLE)_beginthreadex(0, 0, start_wrapper, this, 0, 0); }
    ~Thread() {}
    void join() { WaitForSingleObject(handle, INFINITE); CloseHandle(handle); }
private:
    friend unsigned __stdcall start_wrapper(void* args)
    {
        Thread* t = (Thread*)args;
        t->_start(t->_args);
        return 0;
    }
    HANDLE handle;
    void* (*_start)(void*);
    void* _args;
};

class NCNN_EXPORT ThreadLocalStorage
{
public:
    ThreadLocalStorage() { key = TlsAlloc(); }
    ~ThreadLocalStorage() { TlsFree(key); }
    void set(void* value) { TlsSetValue(key, (LPVOID)value); }
    void* get() { return (void*)TlsGetValue(key); }
private:
    DWORD key;
};
#else // (defined _WIN32 && !(defined __MINGW32__))
class NCNN_EXPORT Mutex
{
public:
    Mutex() { pthread_mutex_init(&mutex, 0); }
    ~Mutex() { pthread_mutex_destroy(&mutex); }
    void lock() { pthread_mutex_lock(&mutex); }
    void unlock() { pthread_mutex_unlock(&mutex); }
private:
    friend class ConditionVariable;
    pthread_mutex_t mutex;
};

class NCNN_EXPORT ConditionVariable
{
public:
    ConditionVariable() { pthread_cond_init(&cond, 0); }
    ~ConditionVariable() { pthread_cond_destroy(&cond); }
    void wait(Mutex& mutex) { pthread_cond_wait(&cond, &mutex.mutex); }
    void broadcast() { pthread_cond_broadcast(&cond); }
    void signal() { pthread_cond_signal(&cond); }
private:
    pthread_cond_t cond;
};

class NCNN_EXPORT Thread
{
public:
    Thread(void* (*start)(void*), void* args = 0) { pthread_create(&t, 0, start, args); }
    ~Thread() {}
    void join() { pthread_join(t, 0); }
private:
    pthread_t t;
};

class NCNN_EXPORT ThreadLocalStorage
{
public:
    ThreadLocalStorage() { pthread_key_create(&key, 0); }
    ~ThreadLocalStorage() { pthread_key_delete(key); }
    void set(void* value) { pthread_setspecific(key, value); }
    void* get() { return pthread_getspecific(key); }
private:
    pthread_key_t key;
};
#endif // (defined _WIN32 && !(defined __MINGW32__))
#else // NCNN_THREADS
class NCNN_EXPORT Mutex
{
public:
    Mutex() {}
    ~Mutex() {}
    void lock() {}
    void unlock() {}
};

class NCNN_EXPORT ConditionVariable
{
public:
    ConditionVariable() {}
    ~ConditionVariable() {}
    void wait(Mutex& /*mutex*/) {}
    void broadcast() {}
    void signal() {}
};

class NCNN_EXPORT Thread
{
public:
    Thread(void* (*/*start*/)(void*), void* /*args*/ = 0) {}
    ~Thread() {}
    void join() {}
};

class NCNN_EXPORT ThreadLocalStorage
{
public:
    ThreadLocalStorage() { data = 0; }
    ~ThreadLocalStorage() {}
    void set(void* value) { data = value; }
    void* get() { return data; }
private:
    void* data;
};
#endif // NCNN_THREADS

class NCNN_EXPORT MutexLockGuard
{
public:
    MutexLockGuard(Mutex& _mutex) : mutex(_mutex) { mutex.lock(); }
    ~MutexLockGuard() { mutex.unlock(); }
private:
    Mutex& mutex;
};

} // namespace ncnn

#if NCNN_SIMPLESTL
#include "simplestl.h"
#else
#include <algorithm>
#include <list>
#include <vector>
#include <string>
#endif

#endif // __cplusplus

#if NCNN_STDIO
#if NCNN_PLATFORM_API && __ANDROID_API__ >= 8
#include <android/log.h>
#define NCNN_LOGE(...) do { \
    fprintf(stderr, ##__VA_ARGS__); fprintf(stderr, "\n"); \
    __android_log_print(ANDROID_LOG_WARN, "ncnn", ##__VA_ARGS__); } while(0)
#else // NCNN_PLATFORM_API && __ANDROID_API__ >= 8
#include <stdio.h>
#define NCNN_LOGE(...) do { \
    fprintf(stderr, ##__VA_ARGS__); fprintf(stderr, "\n"); } while(0)
#endif // NCNN_PLATFORM_API && __ANDROID_API__ >= 8
#else
#define NCNN_LOGE(...)
#endif


#if NCNN_FORCE_INLINE
#ifdef _MSC_VER
    #define NCNN_FORCEINLINE __forceinline
#elif defined(__GNUC__)
    #define NCNN_FORCEINLINE inline __attribute__((__always_inline__))
#elif defined(__CLANG__)
    #if __has_attribute(__always_inline__)
        #define NCNN_FORCEINLINE inline __attribute__((__always_inline__))
    #else
        #define NCNN_FORCEINLINE inline
    #endif
#else
    #define NCNN_FORCEINLINE inline
#endif
#else
    #define NCNN_FORCEINLINE inline
#endif

#endif // NCNN_PLATFORM_H

	// Tencent is pleased to support the open source community by making ncnn available.
	//
	// Copyright (C) 2017 THL A29 Limited, a Tencent company. All rights reserved.
	//
	// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
	// in compliance with the License. You may obtain a copy of the License at
	//
	// https://opensource.org/licenses/BSD-3-Clause
	//
	// 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.

	#ifndef NCNN_PLATFORM_H
	#define NCNN_PLATFORM_H

	#define NCNN_STDIO 1
	#define NCNN_STRING 1
	#define NCNN_SIMPLEOCV 0
	#define NCNN_SIMPLEOMP 0
	#define NCNN_SIMPLESTL 0
	#define NCNN_THREADS 1
	#define NCNN_BENCHMARK 0
	#define NCNN_C_API 1
	#define NCNN_PLATFORM_API 1
	#define NCNN_PIXEL 1
	#define NCNN_PIXEL_ROTATE 1
	#define NCNN_PIXEL_AFFINE 1
	#define NCNN_PIXEL_DRAWING 1
	#define NCNN_VULKAN 0
	#define NCNN_SYSTEM_GLSLANG 0
	#define NCNN_RUNTIME_CPU 1
	#define NCNN_GNU_INLINE_ASM 1
	#define NCNN_AVX 1
	#define NCNN_XOP 1
	#define NCNN_FMA 1
	#define NCNN_F16C 1
	#define NCNN_AVX2 1
	#define NCNN_AVXVNNI 1
	#define NCNN_AVX512 1
	#define NCNN_AVX512VNNI 1
	#define NCNN_AVX512BF16 0
	#define NCNN_AVX512FP16 0
	#define NCNN_VFPV4 0
	#define NCNN_ARM82 0
	#define NCNN_ARM82DOT 0
	#define NCNN_ARM82FP16FML 0
	#define NCNN_ARM84BF16 0
	#define NCNN_ARM84I8MM 0
	#define NCNN_ARM86SVE 0
	#define NCNN_ARM86SVE2 0
	#define NCNN_ARM86SVEBF16 0
	#define NCNN_ARM86SVEI8MM 0
	#define NCNN_ARM86SVEF32MM 0
	#define NCNN_MSA 0
	#define NCNN_LSX 0
	#define NCNN_MMI 0
	#define NCNN_RVV 0
	#define NCNN_INT8 1
	#define NCNN_BF16 1
	#define NCNN_FORCE_INLINE 1

	#define NCNN_VERSION_STRING "1.0.20230913"

	#include "ncnn_export.h"

	#ifdef __cplusplus

	#if NCNN_THREADS
	#if (defined _WIN32 && !(defined __MINGW32__))
	#define WIN32_LEAN_AND_MEAN
	#include <windows.h>
	#include <process.h>
	#else
	#include <pthread.h>
	#endif
	#endif // NCNN_THREADS

	#if __ANDROID_API__ >= 26
	#define VK_USE_PLATFORM_ANDROID_KHR
	#endif // __ANDROID_API__ >= 26

	namespace ncnn {

	#if NCNN_THREADS
	#if (defined _WIN32 && !(defined __MINGW32__))
	class NCNN_EXPORT Mutex
	{
	public:
	Mutex() { InitializeSRWLock(&srwlock); }
	~Mutex() {}
	void lock() { AcquireSRWLockExclusive(&srwlock); }
	void unlock() { ReleaseSRWLockExclusive(&srwlock); }
	private:
	friend class ConditionVariable;
	// NOTE SRWLock is available from windows vista
	SRWLOCK srwlock;
	};

	class NCNN_EXPORT ConditionVariable
	{
	public:
	ConditionVariable() { InitializeConditionVariable(&condvar); }
	~ConditionVariable() {}
	void wait(Mutex& mutex) { SleepConditionVariableSRW(&condvar, &mutex.srwlock, INFINITE, 0); }
	void broadcast() { WakeAllConditionVariable(&condvar); }
	void signal() { WakeConditionVariable(&condvar); }
	private:
	CONDITION_VARIABLE condvar;
	};

	static unsigned __stdcall start_wrapper(void* args);
	class NCNN_EXPORT Thread
	{
	public:
	Thread(void* (start)(void), void* args = 0) { _start = start; _args = args; handle = (HANDLE)_beginthreadex(0, 0, start_wrapper, this, 0, 0); }
	~Thread() {}
	void join() { WaitForSingleObject(handle, INFINITE); CloseHandle(handle); }
	private:
	friend unsigned __stdcall start_wrapper(void* args)
	{
	Thread* t = (Thread*)args;
	t->_start(t->_args);
	return 0;
	}
	HANDLE handle;
	void* (_start)(void);
	void* _args;
	};

	class NCNN_EXPORT ThreadLocalStorage
	{
	public:
	ThreadLocalStorage() { key = TlsAlloc(); }
	~ThreadLocalStorage() { TlsFree(key); }
	void set(void* value) { TlsSetValue(key, (LPVOID)value); }
	void* get() { return (void*)TlsGetValue(key); }
	private:
	DWORD key;
	};
	#else // (defined _WIN32 && !(defined __MINGW32__))
	class NCNN_EXPORT Mutex
	{
	public:
	Mutex() { pthread_mutex_init(&mutex, 0); }
	~Mutex() { pthread_mutex_destroy(&mutex); }
	void lock() { pthread_mutex_lock(&mutex); }
	void unlock() { pthread_mutex_unlock(&mutex); }
	private:
	friend class ConditionVariable;
	pthread_mutex_t mutex;
	};

	class NCNN_EXPORT ConditionVariable
	{
	public:
	ConditionVariable() { pthread_cond_init(&cond, 0); }
	~ConditionVariable() { pthread_cond_destroy(&cond); }
	void wait(Mutex& mutex) { pthread_cond_wait(&cond, &mutex.mutex); }
	void broadcast() { pthread_cond_broadcast(&cond); }
	void signal() { pthread_cond_signal(&cond); }
	private:
	pthread_cond_t cond;
	};

	class NCNN_EXPORT Thread
	{
	public:
	Thread(void* (start)(void), void* args = 0) { pthread_create(&t, 0, start, args); }
	~Thread() {}
	void join() { pthread_join(t, 0); }
	private:
	pthread_t t;
	};

	class NCNN_EXPORT ThreadLocalStorage
	{
	public:
	ThreadLocalStorage() { pthread_key_create(&key, 0); }
	~ThreadLocalStorage() { pthread_key_delete(key); }
	void set(void* value) { pthread_setspecific(key, value); }
	void* get() { return pthread_getspecific(key); }
	private:
	pthread_key_t key;
	};
	#endif // (defined _WIN32 && !(defined __MINGW32__))
	#else // NCNN_THREADS
	class NCNN_EXPORT Mutex
	{
	public:
	Mutex() {}
	~Mutex() {}
	void lock() {}
	void unlock() {}
	};

	class NCNN_EXPORT ConditionVariable
	{
	public:
	ConditionVariable() {}
	~ConditionVariable() {}
	void wait(Mutex& /mutex/) {}
	void broadcast() {}
	void signal() {}
	};

	class NCNN_EXPORT Thread
	{
	public:
	Thread(void* (/start/)(void), void* /args/ = 0) {}
	~Thread() {}
	void join() {}
	};

	class NCNN_EXPORT ThreadLocalStorage
	{
	public:
	ThreadLocalStorage() { data = 0; }
	~ThreadLocalStorage() {}
	void set(void* value) { data = value; }
	void* get() { return data; }
	private:
	void* data;
	};
	#endif // NCNN_THREADS

	class NCNN_EXPORT MutexLockGuard
	{
	public:
	MutexLockGuard(Mutex& _mutex) : mutex(_mutex) { mutex.lock(); }
	~MutexLockGuard() { mutex.unlock(); }
	private:
	Mutex& mutex;
	};

	} // namespace ncnn

	#if NCNN_SIMPLESTL
	#include "simplestl.h"
	#else
	#include <algorithm>
	#include <list>
	#include <vector>
	#include <string>
	#endif

	#endif // __cplusplus

	#if NCNN_STDIO
	#if NCNN_PLATFORM_API && __ANDROID_API__ >= 8
	#include <android/log.h>
	#define NCNN_LOGE(...) do { \
	fprintf(stderr, ##__VA_ARGS__); fprintf(stderr, "\n"); \
	__android_log_print(ANDROID_LOG_WARN, "ncnn", ##__VA_ARGS__); } while(0)
	#else // NCNN_PLATFORM_API && __ANDROID_API__ >= 8
	#include <stdio.h>
	#define NCNN_LOGE(...) do { \
	fprintf(stderr, ##__VA_ARGS__); fprintf(stderr, "\n"); } while(0)
	#endif // NCNN_PLATFORM_API && __ANDROID_API__ >= 8
	#else
	#define NCNN_LOGE(...)
	#endif


	#if NCNN_FORCE_INLINE
	#ifdef _MSC_VER
	#define NCNN_FORCEINLINE __forceinline
	#elif defined(__GNUC__)
	#define NCNN_FORCEINLINE inline __attribute__((__always_inline__))
	#elif defined(__CLANG__)
	#if __has_attribute(__always_inline__)
	#define NCNN_FORCEINLINE inline __attribute__((__always_inline__))
	#else
	#define NCNN_FORCEINLINE inline
	#endif
	#else
	#define NCNN_FORCEINLINE inline
	#endif
	#else
	#define NCNN_FORCEINLINE inline
	#endif

	#endif // NCNN_PLATFORM_H