Sierra Toolkit  Version of the Day
CudaDeviceMgr.hpp
1 /*------------------------------------------------------------------------*/
2 /* Copyright 2010 Sandia Corporation. */
3 /* Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive */
4 /* license for use of this work by or on behalf of the U.S. Government. */
5 /* Export of this program may require a license from the */
6 /* United States Government. */
7 /*------------------------------------------------------------------------*/
8 
9 #ifndef stk_algsup_CudaDeviceMgr_hpp
10 #define stk_algsup_CudaDeviceMgr_hpp
11 
12 #ifdef STK_HAVE_CUDA
13 
14 #include <stk_algsup/CudaCall.hpp>
15 
16 namespace stk_classic {
17 
18 class CudaDeviceMgr {
19  public:
20  CudaDeviceMgr(int device=0);
21 
22  virtual ~CudaDeviceMgr() {}
23 
24  int get_device() const { return m_device; }
25 
26  static CudaDeviceMgr& get_singleton();
27 
28  private:
29  int m_device;
30 };//class CudaMemoryMgr
31 
32 }//namespace stk_classic
33 
34 #endif
35 
36 #endif
37 
Sierra Toolkit.