commit 502aaf39cf4a878f9e4f849e5f409573aa598aa9
author Zach DeVito <zdevito@fb.com> Thu Nov 02 16:26:19 2017 -0700
committer Edward Z. Yang <ezyang@mit.edu> Thu Nov 02 19:53:36 2017 -0400
tree c34d64a75731e54d9699be16129ac9c06640b096
parent 81e56ff8aaa56fc98ee3574777c873a14b60ae43

make sure stdatomic.h is included when checking for ATOMIC_INT_LOCK_FREE

aten/src/TH/THAllocator.c

diff --git a/aten/src/TH/THAllocator.c b/aten/src/TH/THAllocator.c
index 5954949..ea42d11f 100644
--- a/aten/src/TH/THAllocator.c
+++ b/aten/src/TH/THAllocator.c
@@ -1,6 +1,18 @@
 #include "THAllocator.h"
 #include "THAtomic.h"
 
+/* needed for ATOMIC_INT_LOCK_FREE */
+/* cannot go in THAtomic.h because of interactions with OpenMP giving
+   sorry not implemented errors */
+#if defined(USE_C11_ATOMICS)
+#include <stdatomic.h>
+#if ATOMIC_INT_LOCK_FREE == 2
+#define TH_ATOMIC_IPC_REFCOUNT 1
+#endif
+#elif defined(USE_MSC_ATOMICS) || defined(USE_GCC_ATOMICS)
+#define TH_ATOMIC_IPC_REFCOUNT 1
+#endif
+
 /* stuff for mapped files */
 #ifdef _WIN32
 #include <windows.h>

aten/src/TH/THAtomic.h

diff --git a/aten/src/TH/THAtomic.h b/aten/src/TH/THAtomic.h
index 24c43d3..8431664 100644
--- a/aten/src/TH/THAtomic.h
+++ b/aten/src/TH/THAtomic.h
@@ -115,11 +115,4 @@
 */
 TH_API ptrdiff_t THAtomicCompareAndSwapPtrdiff(ptrdiff_t volatile *a, ptrdiff_t oldvalue, ptrdiff_t newvalue);
 
-#if defined(USE_C11_ATOMICS) && defined(ATOMIC_INT_LOCK_FREE) && \
-  ATOMIC_INT_LOCK_FREE == 2
-#define TH_ATOMIC_IPC_REFCOUNT 1
-#elif defined(USE_MSC_ATOMICS) || defined(USE_GCC_ATOMICS)
-#define TH_ATOMIC_IPC_REFCOUNT 1
-#endif
-
 #endif