Assume.h 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. /*
  2. * Copyright 2015-present Facebook, Inc.
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. #pragma once
  17. #include <cstdlib>
  18. #include <folly/Portability.h>
  19. namespace folly {
  20. namespace detail {
  21. extern void assume_check(bool cond);
  22. } // namespace detail
  23. /**
  24. * Inform the compiler that the argument can be assumed true. It is
  25. * undefined behavior if the argument is not actually true, so use
  26. * with care.
  27. *
  28. * Implemented as a function instead of a macro because
  29. * __builtin_assume does not evaluate its argument at runtime, so it
  30. * cannot be used with expressions that have side-effects.
  31. */
  32. FOLLY_ALWAYS_INLINE void assume(bool cond) {
  33. if (kIsDebug) {
  34. detail::assume_check(cond);
  35. } else {
  36. #if defined(__clang__) // Must go first because Clang also defines __GNUC__.
  37. __builtin_assume(cond);
  38. #elif defined(__GNUC__)
  39. if (!cond) {
  40. __builtin_unreachable();
  41. }
  42. #elif defined(_MSC_VER)
  43. __assume(cond);
  44. #else
  45. // Do nothing.
  46. #endif
  47. }
  48. }
  49. [[noreturn]] FOLLY_ALWAYS_INLINE void assume_unreachable() {
  50. assume(false);
  51. // Do a bit more to get the compiler to understand
  52. // that this function really will never return.
  53. #if defined(__GNUC__)
  54. __builtin_unreachable();
  55. #elif defined(_MSC_VER)
  56. __assume(0);
  57. #else
  58. // Well, it's better than nothing.
  59. std::abort();
  60. #endif
  61. }
  62. } // namespace folly