ATLAS Offline Software
Loading...
Searching...
No Matches
SingleStreamSvc.cxx
Go to the documentation of this file.
1//
2// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
3//
4
5// Local include(s).
6#include "SingleStreamSvc.h"
7
8// System include(s).
9#include <cassert>
10
11namespace AthCUDA {
12
14
15 // Create the stream object.
16 m_stream = std::make_unique<const Details::Stream>();
17
18 // Tell the user what happened.
19 ATH_MSG_INFO("Initialized CUDA stream on device: " << m_stream->name());
20
21 // Return gracefully.
22 return StatusCode::SUCCESS;
23}
24
25cudaStream_t SingleStreamSvc::stream(const EventContext&) const {
26
27 // Get the stream corresponding to the current slot.
28 assert(m_stream);
29 assert(m_stream->stream());
30 return m_stream->stream();
31}
32
33} // namespace AthCUDA
#define ATH_MSG_INFO(x)
virtual StatusCode initialize() override
Initialize the tool.
std::unique_ptr< const Details::Stream > m_stream
Single CUDA stream.
virtual cudaStream_t stream(const EventContext &ctx) const override
Get the CUDA stream to use.
Definition Info.h:14