LinearTransformAsyncExampleAlg.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
//
// Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHEXCUDA_LINEARTRANSFORMASYNCEXAMPLEALG_H
#define ATHEXCUDA_LINEARTRANSFORMASYNCEXAMPLEALG_H
 
// Framework include(s).
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAsynchronousAlgorithm.h"
 
// CUDA include(s).
#include <cuda_runtime_api.h>
 
// STL
#include <vector>
 
namespace AthCUDAExamples {

   class LinearTransformAsyncExampleAlg : public AthAsynchronousAlgorithm {
 
   public:
      // Inherit the base class's constructor(s).
      using AthAsynchronousAlgorithm::AthAsynchronousAlgorithm;

      virtual StatusCode execute( const EventContext& ctx ) const override;
 
   }; // class LinearTransformAsyncExampleAlg

   void linearTransform(cudaStream_t stream, std::vector<float>& arr, float multiplier);
 
} // namespace AthCUDAExamples
 
#endif // ATHEXCUDA_LINEARTRANSFORMASYNCEXAMPLEALG_H