Use shared_mutex and shared_lock

Signed-off-by: Nghia Truong <[email protected]>

cpp/include/cudf/utilities/prefetch.hpp

@@ -21,7 +21,7 @@
 #include <rmm/device_uvector.hpp>
 
 #include <map>
-#include <mutex>
+#include <shared_mutex>
 #include <string>
 #include <string_view>
 
@@ -53,7 +53,7 @@ class prefetch_config {
    * @param key The configuration key.
    * @return The value of the configuration key.
    */
-  bool get(std::string_view key) const;
+  bool get(std::string_view key);
   /**
    * @brief Set the value of a configuration key.
    *
@@ -73,7 +73,7 @@ class prefetch_config {
  private:
   prefetch_config() = default;                //< Private constructor to enforce singleton pattern
   std::map<std::string, bool> config_values;  //< Map of configuration keys to values
-  std::mutex config_mtx;                      //< Mutex for thread-safe config access
+  std::shared_mutex config_mtx;               //< Mutex for thread-safe config access
 };
 
 /**

cpp/src/utilities/prefetch.cpp

@@ -32,17 +32,18 @@ prefetch_config& prefetch_config::instance()
   return instance;
 }
 
-bool prefetch_config::get(std::string_view key) const
+bool prefetch_config::get(std::string_view key)
 {
-  std::scoped_lock lock(config_mtx);
-  // Default to not prefetching
-  if (config_values.find(key.data()) == config_values.end()) { return false; }
+  std::shared_lock<std::shared_mutex> lock(config_mtx);
+  if (config_values.find(key.data()) == config_values.end()) {
+    return false;  // default to not prefetching
+  }
   return config_values.at(key.data());
 }
 
 void prefetch_config::set(std::string_view key, bool value)
 {
-  std::scoped_lock lock(config_mtx);
+  std::lock_guard<std::shared_mutex> lock(config_mtx);
   config_values[key.data()] = value;
 }