detail/stream.hpp

/*
 * SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
 
#include <map>
#include <mutex>
#include <thread>
 
#include <kvikio/shim/cuda.hpp>
 
namespace kvikio::detail {
class StreamCachePerThreadAndContext {
 private:
  std::map<std::pair<CUcontext, std::thread::id>, CUstream> _streams;
  std::mutex mutable _mutex;
 
 private:
  StreamCachePerThreadAndContext()  = default;
  ~StreamCachePerThreadAndContext() = default;
 
 public:
  KVIKIO_EXPORT static CUstream get();
 
  StreamCachePerThreadAndContext(StreamCachePerThreadAndContext const&)            = delete;
  StreamCachePerThreadAndContext& operator=(StreamCachePerThreadAndContext const&) = delete;
  StreamCachePerThreadAndContext(StreamCachePerThreadAndContext&& o)               = delete;
  StreamCachePerThreadAndContext& operator=(StreamCachePerThreadAndContext&& o)    = delete;
};
}  // namespace kvikio::detail