GPUMLib  0.2.2
GPU Machine Learning Library
CudaStreams.h
1 /*
2  Noel Lopes is an Assistant Professor at the Polytechnic Institute of Guarda, Portugal
3  Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014 Noel de Jesus Mendonša Lopes
4 
5  This file is part of GPUMLib.
6 
7  GPUMLib is free software: you can redistribute it and/or modify
8  it under the terms of the GNU General Public License as published by
9  the Free Software Foundation, either version 3 of the License, or
10  (at your option) any later version.
11 
12  This program is distributed in the hope that it will be useful,
13  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15  GNU General Public License for more details.
16 
17  You should have received a copy of the GNU General Public License
18  along with this program. If not, see <http://www.gnu.org/licenses/>.
19 */
20 
21 #ifndef GPUMLib_CudaStreams_h
22 #define GPUMLib_CudaStreams_h
23 
24 #include <cuda_runtime.h>
25 
26 namespace GPUMLib {
27 
30 
32 class CudaStream {
33  private:
34  cudaStream_t stream;
35 
36  public:
39  cudaStreamCreate(&stream);
40  }
41 
44  cudaStreamDestroy(stream);
45  }
46 
48  operator cudaStream_t () {
49  return stream;
50  }
51 };
52 
54 
55 }
56 
57 #endif
CudaStream()
Constructor.
Definition: CudaStreams.h:38
Represents a CUDA stream.
Definition: CudaStreams.h:32
~CudaStream()
Destructor.
Definition: CudaStreams.h:43