GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-06-04 18:54:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support per-thread is_malloc_allowed() state

Browse Source
This commit is contained in:
tttapa 2023-01-16 01:34:56 +00:00 committed by Rasmus Munk Larsen
parent fa0bd2c34e
commit bae119bb7e
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
Eigen/src/Core/util/Memory.h
View File

@ -59,6 +59,25 @@
#endif #endif
#ifndef EIGEN_MALLOC_CHECK_THREAD_LOCAL
// Check whether we can use the thread_local keyword to allow or disallow
// allocating memory with per-thread granularity, by means of the
// set_is_malloc_allowed() function.
#ifndef EIGEN_AVOID_THREAD_LOCAL
#if ((EIGEN_COMP_GNUC) || __has_feature(cxx_thread_local) || EIGEN_COMP_MSVC >= 1900) && !defined(EIGEN_GPU_COMPILE_PHASE)
#define EIGEN_MALLOC_CHECK_THREAD_LOCAL thread_local
#else
#define EIGEN_MALLOC_CHECK_THREAD_LOCAL
#endif
#else // EIGEN_AVOID_THREAD_LOCAL
#define EIGEN_MALLOC_CHECK_THREAD_LOCAL
#endif // EIGEN_AVOID_THREAD_LOCAL
#endif
#include "../InternalHeaderCheck.h" #include "../InternalHeaderCheck.h"
namespace Eigen { namespace Eigen {
@ -156,7 +175,7 @@ EIGEN_DEVICE_FUNC inline void check_that_malloc_is_allowed()
#elif defined EIGEN_RUNTIME_NO_MALLOC #elif defined EIGEN_RUNTIME_NO_MALLOC
EIGEN_DEVICE_FUNC inline bool is_malloc_allowed_impl(bool update, bool new_value = false) EIGEN_DEVICE_FUNC inline bool is_malloc_allowed_impl(bool update, bool new_value = false)
{ {
static bool value = true; EIGEN_MALLOC_CHECK_THREAD_LOCAL static bool value = true;
if (update == 1) if (update == 1)
value = new_value; value = new_value;
return value; return value;