Support thread-safe for prefetch_config::get and prefetch_config::set

cpp/include/cudf/utilities/prefetch.hpp

@@ -21,6 +21,7 @@
 #include <rmm/device_uvector.hpp>
 
 #include <map>
+#include <mutex>
 #include <string>
 #include <string_view>
 
@@ -47,13 +48,17 @@ class prefetch_config {
   /**
    * @brief Get the value of a configuration key.
    *
+   * If the key does not exist, a `false` value will be returned.
+   *
    * @param key The configuration key.
    * @return The value of the configuration key.
    */
-  bool get(std::string_view key);
+  bool get(std::string_view key) const;
   /**
    * @brief Set the value of a configuration key.
    *
+   * This is a thread-safe operation.
+   *
    * @param key The configuration key.
    * @param value The value to set.
    */
@@ -68,6 +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
 };
 
 /**

cpp/src/utilities/prefetch.cpp

@@ -32,15 +32,18 @@ prefetch_config& prefetch_config::instance()
   return instance;
 }
 
-bool prefetch_config::get(std::string_view key)
+bool prefetch_config::get(std::string_view key) const
 {
   // Default to not prefetching
-  if (config_values.find(key.data()) == config_values.end()) {
-    return (config_values[key.data()] = false);
-  }
-  return config_values[key.data()];
+  if (config_values.find(key.data()) == config_values.end()) { return false; }
+  return config_values.get(key.data());
+}
+
+void prefetch_config::set(std::string_view key, bool value)
+{
+  std::scoped_lock lock(config_mtx);
+  config_values[key.data()] = value;
 }
-void prefetch_config::set(std::string_view key, bool value) { config_values[key.data()] = value; }
 
 cudaError_t prefetch_noexcept(std::string_view key,
                               void const* ptr,