StreamWrapper.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file StreamWrapper.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

// CUDA plugin include(s).
#include "Acts/Plugins/Cuda/Utilities/Info.hpp"

namespace Acts {
namespace Cuda {

class StreamWrapper {
#ifdef __CUDACC__
  friend cudaStream_t getStreamFrom(const StreamWrapper&);
#endif  // __CUDACC__

 public:
  StreamWrapper(void* stream, bool ownsStream = true);
  StreamWrapper(StreamWrapper&& parent);
  StreamWrapper(const StreamWrapper&) = delete;
  ~StreamWrapper();

  StreamWrapper& operator=(StreamWrapper&& rhs);
  StreamWrapper& operator=(const StreamWrapper&) = delete;

  void synchronize() const;

 private:
  void* m_stream;
  bool m_ownsStream;

};  // class StreamWrapper

StreamWrapper createStreamFor(const Acts::Cuda::Info::Device& device);

}  // namespace Cuda
}  // namespace Acts