SHOGUN  4.1.0
 全部  命名空间 文件 函数 变量 类型定义 枚举 枚举值 友元 宏定义  
GPUVector.h
浏览该文件的文档.
1 /*
2  * Copyright (c) 2014, Shogun Toolbox Foundation
3  * All rights reserved.
4  *
5  * Written (W) 2014 Khaled Nasr
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions are met:
9 
10  * 1. Redistributions of source code must retain the above copyright notice,
11  * this list of conditions and the following disclaimer.
12  *
13  * 2. Redistributions in binary form must reproduce the above copyright notice,
14  * this list of conditions and the following disclaimer in the documentation
15  * and/or other materials provided with the distribution.
16  *
17  * 3. Neither the name of the copyright holder nor the names of its
18  * contributors may be used to endorse or promote products derived from this
19  * software without specific prior written permission.
20 
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
25  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
31  * POSSIBILITY OF SUCH DAMAGE.
32  */
33 
34 #ifndef __GPUVECTOR_H__
35 #define __GPUVECTOR_H__
36 
37 #include <shogun/lib/config.h>
38 
39 #ifdef HAVE_VIENNACL
40 #ifdef HAVE_CXX11
41 
42 #include <shogun/lib/common.h>
43 #include <memory>
44 
45 #ifndef SWIG // SWIG should skip this part
46 namespace viennacl
47 {
48  template <class, class, class> class vector_base;
49  template <class> class const_entry_proxy;
50  template <class> class entry_proxy;
51 
52  namespace backend
53  {
54  class mem_handle;
55  }
56 }
57 
58 namespace Eigen
59 {
60  template <class, int, int, int, int, int> class Matrix;
61 }
62 
63 namespace shogun
64 {
65 
77 template <class T> class CGPUVector
78 {
79 
80  typedef viennacl::vector_base<T, std::size_t, std::ptrdiff_t> VCLVectorBase;
81  typedef viennacl::backend::mem_handle VCLMemoryArray;
82 
83  typedef Eigen::Matrix<T,-1,1,0,-1,1> EigenVectorXt;
84  typedef Eigen::Matrix<T,1,-1,0x1,1,-1> EigenRowVectorXt;
85 
86 public:
88  typedef T Scalar;
89 
91  template <typename ST> using container_type = CGPUVector<ST>;
92 
94  CGPUVector();
95 
100  CGPUVector(index_t length);
101 
109  CGPUVector(std::shared_ptr<VCLMemoryArray> mem, index_t length, index_t mem_offset=0);
110 
112  CGPUVector(const SGVector<T>& cpu_vec);
113 
114 #ifdef HAVE_EIGEN3
115 
116  CGPUVector(const EigenVectorXt& cpu_vec);
117 
119  CGPUVector(const EigenRowVectorXt& cpu_vec);
120 
122  operator EigenVectorXt() const;
123 
125  operator EigenRowVectorXt() const;
126 
127 #endif // HAVE_EIGEN3
128 
130  inline VCLVectorBase data()
131  {
132  return vcl_vector();
133  }
134 
136  inline index_t size() const
137  {
138  return vlen;
139  }
140 
142  operator SGVector<T>() const;
143 
147  VCLVectorBase vcl_vector();
148 
150  void zero();
151 
156  void set_const(T value);
157 
159  void display_vector(const char* name="vector") const;
160 
166  viennacl::const_entry_proxy<T> operator[](index_t index) const;
167 
173  viennacl::entry_proxy<T> operator[](index_t index);
174 
175 private:
176  void init();
177 
178 public:
180  std::shared_ptr<VCLMemoryArray> vector;
181 
185  index_t offset;
186 
188  index_t vlen;
189 };
190 
191 }
192 #endif // SWIG
193 
194 #endif // HAVE_CXX11
195 #endif // HAVE_VIENNACL
196 #endif // __GPUVECTOR_H__
int32_t index_t
Definition: common.h:62
Definition: SGMatrix.h:20
all of classes and functions are contained in the shogun namespace
Definition: class_list.h:18

SHOGUN 机器学习工具包 - 项目文档