fix: source location as argument of check function

kokkos · Jul 29, 2024 · d8ee84a · d8ee84a
1 parent e205792
commit d8ee84a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/common/src/KokkosFFT_utils.hpp b/common/src/KokkosFFT_utils.hpp
@@ -15,8 +15,9 @@
 
 #ifndef KOKKOS_ENABLE_CXX17
 #include <source_location>
-#define KOKKOSFFT_EXPECTS(expression, msg) \
-  KokkosFFT::Impl::check_precondition(expression, msg)
+#define KOKKOSFFT_EXPECTS(expression, msg)             \
+  KokkosFFT::Impl::check_precondition(expression, msg, \
+                                      std::source_location::current())
 #else
 #include <cstdlib>
 #define KOKKOSFFT_EXPECTS(expression, msg)                                 \
@@ -28,9 +29,9 @@ namespace KokkosFFT {
 namespace Impl {
 
 #ifndef KOKKOS_ENABLE_CXX17
-inline void check_precondition(const bool expression, const std::string& msg) {
+inline void check_precondition(const bool expression, const std::string& msg,
+                               std::source_location location) {
   std::stringstream ss("file: ");
-  std::source_location location = std::source_location::current();
   ss << location.file_name() << '(' << location.line() << ':'
      << location.column() << ") `" << location.function_name() << "`: " << msg
      << '\n';