atomic_ops.h 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. /*
  2. * Original code from jemalloc with this license:
  3. *
  4. * Copyright (C) 2002-2013 Jason Evans <jasone@canonware.com>.
  5. * All rights reserved.
  6. * Copyright (C) 2007-2012 Mozilla Foundation. All rights reserved.
  7. * Copyright (C) 2009-2013 Facebook, Inc. All rights reserved.
  8. *
  9. * Redistribution and use in source and binary forms, with or without
  10. * modification, are permitted provided that the following conditions are met:
  11. * 1. Redistributions of source code must retain the above copyright notice(s),
  12. * this list of conditions and the following disclaimer.
  13. * 2. Redistributions in binary form must reproduce the above copyright notice(s),
  14. * this list of conditions and the following disclaimer in the documentation
  15. * and/or other materials provided with the distribution.
  16. *
  17. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER(S) ``AS IS'' AND ANY EXPRESS
  18. * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  19. * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
  20. * EVENT SHALL THE COPYRIGHT HOLDER(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
  21. * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22. * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  23. * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  24. * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
  25. * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  26. * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27. * This program is free software; you can redistribute it and/or
  28. * modify it under the terms of the GNU General Public License
  29. * as published by the Free Software Foundation; either version 2
  30. * of the License, or (at your option) any later version.
  31. *
  32. * This program is distributed in the hope that it will be useful,
  33. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  34. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  35. * GNU General Public License for more details.
  36. *
  37. * You should have received a copy of the GNU General Public License
  38. * along with this program; if not, write to the Free Software Foundation,
  39. * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  40. *
  41. * The Original Code is Copyright (C) 2016 Blender Foundation.
  42. * All rights reserved.
  43. *
  44. * The Original Code is: adapted from jemalloc.
  45. */
  46. /** \file
  47. * \ingroup Atomic
  48. *
  49. * \brief Provides wrapper around system-specific atomic primitives,
  50. * and some extensions (faked-atomic operations over float numbers).
  51. */
  52. #ifndef __ATOMIC_OPS_H__
  53. #define __ATOMIC_OPS_H__
  54. #if defined(__arm__)
  55. /* Attempt to fix compilation error on Debian armel kernel.
  56. * arm7 architecture does have both 32 and 64bit atomics, however
  57. * it's gcc doesn't have __GCC_HAVE_SYNC_COMPARE_AND_SWAP_n defined.
  58. */
  59. # define JE_FORCE_SYNC_COMPARE_AND_SWAP_1
  60. # define JE_FORCE_SYNC_COMPARE_AND_SWAP_4
  61. # define JE_FORCE_SYNC_COMPARE_AND_SWAP_8
  62. #endif
  63. #include "intern/atomic_ops_utils.h"
  64. /******************************************************************************/
  65. /* Function prototypes. */
  66. #if (LG_SIZEOF_PTR == 8 || LG_SIZEOF_INT == 8)
  67. ATOMIC_INLINE uint64_t atomic_add_and_fetch_uint64(uint64_t *p, uint64_t x);
  68. ATOMIC_INLINE uint64_t atomic_sub_and_fetch_uint64(uint64_t *p, uint64_t x);
  69. ATOMIC_INLINE uint64_t atomic_fetch_and_add_uint64(uint64_t *p, uint64_t x);
  70. ATOMIC_INLINE uint64_t atomic_fetch_and_sub_uint64(uint64_t *p, uint64_t x);
  71. ATOMIC_INLINE uint64_t atomic_cas_uint64(uint64_t *v, uint64_t old, uint64_t _new);
  72. ATOMIC_INLINE int64_t atomic_add_and_fetch_int64(int64_t *p, int64_t x);
  73. ATOMIC_INLINE int64_t atomic_sub_and_fetch_int64(int64_t *p, int64_t x);
  74. ATOMIC_INLINE int64_t atomic_fetch_and_add_int64(int64_t *p, int64_t x);
  75. ATOMIC_INLINE int64_t atomic_fetch_and_sub_int64(int64_t *p, int64_t x);
  76. ATOMIC_INLINE int64_t atomic_cas_int64(int64_t *v, int64_t old, int64_t _new);
  77. #endif
  78. ATOMIC_INLINE uint32_t atomic_add_and_fetch_uint32(uint32_t *p, uint32_t x);
  79. ATOMIC_INLINE uint32_t atomic_sub_and_fetch_uint32(uint32_t *p, uint32_t x);
  80. ATOMIC_INLINE uint32_t atomic_cas_uint32(uint32_t *v, uint32_t old, uint32_t _new);
  81. ATOMIC_INLINE uint32_t atomic_fetch_and_add_uint32(uint32_t *p, uint32_t x);
  82. ATOMIC_INLINE uint32_t atomic_fetch_and_or_uint32(uint32_t *p, uint32_t x);
  83. ATOMIC_INLINE uint32_t atomic_fetch_and_and_uint32(uint32_t *p, uint32_t x);
  84. ATOMIC_INLINE int32_t atomic_add_and_fetch_int32(int32_t *p, int32_t x);
  85. ATOMIC_INLINE int32_t atomic_sub_and_fetch_int32(int32_t *p, int32_t x);
  86. ATOMIC_INLINE int32_t atomic_cas_int32(int32_t *v, int32_t old, int32_t _new);
  87. ATOMIC_INLINE int32_t atomic_fetch_and_add_int32(int32_t *p, int32_t x);
  88. ATOMIC_INLINE int32_t atomic_fetch_and_or_int32(int32_t *p, int32_t x);
  89. ATOMIC_INLINE int32_t atomic_fetch_and_and_int32(int32_t *p, int32_t x);
  90. ATOMIC_INLINE uint8_t atomic_fetch_and_or_uint8(uint8_t *p, uint8_t b);
  91. ATOMIC_INLINE uint8_t atomic_fetch_and_and_uint8(uint8_t *p, uint8_t b);
  92. ATOMIC_INLINE int8_t atomic_fetch_and_or_int8(int8_t *p, int8_t b);
  93. ATOMIC_INLINE int8_t atomic_fetch_and_and_int8(int8_t *p, int8_t b);
  94. ATOMIC_INLINE char atomic_fetch_and_or_char(char *p, char b);
  95. ATOMIC_INLINE char atomic_fetch_and_and_char(char *p, char b);
  96. ATOMIC_INLINE size_t atomic_add_and_fetch_z(size_t *p, size_t x);
  97. ATOMIC_INLINE size_t atomic_sub_and_fetch_z(size_t *p, size_t x);
  98. ATOMIC_INLINE size_t atomic_fetch_and_add_z(size_t *p, size_t x);
  99. ATOMIC_INLINE size_t atomic_fetch_and_sub_z(size_t *p, size_t x);
  100. ATOMIC_INLINE size_t atomic_cas_z(size_t *v, size_t old, size_t _new);
  101. /* Uses CAS loop, see warning below. */
  102. ATOMIC_INLINE size_t atomic_fetch_and_update_max_z(size_t *p, size_t x);
  103. ATOMIC_INLINE unsigned int atomic_add_and_fetch_u(unsigned int *p, unsigned int x);
  104. ATOMIC_INLINE unsigned int atomic_sub_and_fetch_u(unsigned int *p, unsigned int x);
  105. ATOMIC_INLINE unsigned int atomic_fetch_and_add_u(unsigned int *p, unsigned int x);
  106. ATOMIC_INLINE unsigned int atomic_fetch_and_sub_u(unsigned int *p, unsigned int x);
  107. ATOMIC_INLINE unsigned int atomic_cas_u(unsigned int *v, unsigned int old, unsigned int _new);
  108. ATOMIC_INLINE void *atomic_cas_ptr(void **v, void *old, void *_new);
  109. ATOMIC_INLINE float atomic_cas_float(float *v, float old, float _new);
  110. /* WARNING! Float 'atomics' are really faked ones, those are actually closer to some kind of
  111. * spinlock-sync'ed operation, which means they are only efficient if collisions are highly
  112. * unlikely (i.e. if probability of two threads working on the same pointer at the same time is
  113. * very low). */
  114. ATOMIC_INLINE float atomic_add_and_fetch_fl(float *p, const float x);
  115. /******************************************************************************/
  116. /* Include system-dependent implementations. */
  117. /* Note that we are using _unix flavor as fallback here
  118. * (it will raise precompiler errors as needed). */
  119. #if defined(_MSC_VER)
  120. # include "intern/atomic_ops_msvc.h"
  121. #else
  122. # include "intern/atomic_ops_unix.h"
  123. #endif
  124. /* Include 'fake' atomic extensions, built over real atomic primitives. */
  125. #include "intern/atomic_ops_ext.h"
  126. #endif /* __ATOMIC_OPS_H__ */