Info.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
//
// Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHCUDACORE_INFO_H
#define ATHCUDACORE_INFO_H
 
// System include(s).
#include <cstddef>
#include <iosfwd>
#include <string>
#include <vector>
 
namespace AthCUDA {

   class Info {
 
   public:
      static const Info& instance();

      int nDevices() const;

      const std::vector< std::string >& names() const;
      const std::vector< int >& maxThreadsPerBlock() const;
      const std::vector< bool >& concurrentKernels() const;
      const std::vector< std::size_t >& totalMemory() const;

      void print() const;
 
   private:
      Info();

      int m_nDevices = 0;
      std::vector< std::string > m_names;
      std::vector< int > m_maxThreadsPerBlock;
      std::vector< bool > m_concurrentKernels;
      std::vector< std::size_t > m_totalMemory;
 
   }; // class Info

   std::ostream& operator<< ( std::ostream& out, const AthCUDA::Info& info );
 
} // namespace AthCUDA
 
#endif // ATHCUDACORE_INFO_H