ATLAS Offline Software
Loading...
Searching...
No Matches
Control
AthCUDA
AthCUDAServices
src
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
11
namespace
AthCUDA
{
12
13
StatusCode
SingleStreamSvc::initialize
() {
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
25
cudaStream_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
ATH_MSG_INFO
#define ATH_MSG_INFO(x)
Definition
AthMsgStreamMacros.h:31
SingleStreamSvc.h
AthCUDA::SingleStreamSvc::initialize
virtual StatusCode initialize() override
Initialize the tool.
Definition
SingleStreamSvc.cxx:13
AthCUDA::SingleStreamSvc::m_stream
std::unique_ptr< const Details::Stream > m_stream
Single CUDA stream.
Definition
SingleStreamSvc.h:53
AthCUDA::SingleStreamSvc::stream
virtual cudaStream_t stream(const EventContext &ctx) const override
Get the CUDA stream to use.
Definition
SingleStreamSvc.cxx:25
AthCUDA
Definition
Info.h:14
Generated on
for ATLAS Offline Software by
1.16.1