mutex.hpp

Go to the documentation of this file.

// Copyright 2020 Lawrence Livermore National Security, LLC and other Metall
// Project Developers. See the top-level COPYRIGHT file for details.
//
// SPDX-License-Identifier: (Apache-2.0 OR MIT)
 
#ifndef METALL_UTILITY_MUTEX_HPP
#define METALL_UTILITY_MUTEX_HPP
 
#include <mutex>
#include <cassert>
 
namespace metall::utility {
 
namespace mutex {
 
template <int num_banks>
inline std::unique_lock<std::mutex> mutex_lock(const std::size_t index) {
  static std::mutex mutexes[num_banks];
  assert(index < num_banks);
  return std::unique_lock<std::mutex>(mutexes[index]);
}
 
}  // namespace mutex
}  // namespace metall::utility
 
 
#endif  // METALL_UTILITY_MUTEX_HPP