HeterogeneousCore/AlpakaInterface/interface/host.h - master - githubmirror/cmssw

HeterogeneousCore/AlpakaInterface/interface/host.h

18 строк597 B

Andrea Bocci

Update Alpaka to version 1.1.0

12 фев 2024, 13:29

12 фев 2024, 13:299bfce46

#ifndef HeterogeneousCore_AlpakaInterface_interface_host_h
#define HeterogeneousCore_AlpakaInterface_interface_host_h

#include <alpaka/alpaka.hpp>

#include "HeterogeneousCore/AlpakaInterface/interface/devices.h"

namespace cms::alpakatools {

  // returns the alpaka host platform
  inline alpaka::PlatformCpu const& host_platform() { return platform<alpaka::PlatformCpu>(); }

  // returns the alpaka host device
  inline alpaka::DevCpu const& host() { return devices<alpaka::PlatformCpu>()[0]; }

}  // namespace cms::alpakatools

#endif  // HeterogeneousCore_AlpakaInterface_interface_host_h