arrayfire.github.io/docs/benchmarks_2cg_8cpp-example.htm at master · arrayfire/arrayfire.github.io · GitHub

History

261 lines (261 loc) · 25.6 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">

<head>

window.dataLayer = window.dataLayer || [];

function gtag(){dataLayer.push(arguments);}

gtag('js', new Date());

gtag('config', 'UA-130950618-1');

</script>

<title>ArrayFire: benchmarks/cg.cpp</title>

MathJax.Hub.Config({

extensions: ["tex2jax.js"],

jax: ["input/TeX","output/HTML-CSS"],

});

</script>

DoxygenAwesomeDarkModeToggle.init()

DoxygenAwesomeInteractiveToc.init()

DoxygenAwesomeFragmentCopyButton.init()

</script>

</head>

<body>

<tbody>

</tr>

<td>

<div id="projectbrief">A high-performance general-purpose compute library</div>

</td>

</tr>

<div>

</td>

</div>

</tr>

<tr>

<td>

</td>

</tr>

</tbody>

</table>

</div>

<li><a href="tutorials.htm"><span>Tutorials</span></a></li>

<li><a href="modules.htm"><span>Functions</span></a></li>

<li><a href="releasenotes.htm"><span>Release Notes</span></a></li>

<li><a href="examples.htm"><span>Examples</span></a></li>

</ul>

</div>

</div>

</div>

<div id="splitbar" style="-moz-user-select:none;"

class="ui-resizable-handle">

</div>

/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */

$(document).ready(function(){initNavTree('benchmarks_2cg_8cpp-example.htm',''); initResizable(); });

/* @license-end */

</script>

<div class="headertitle"><div class="title">benchmarks/cg.cpp</div></div>

</div>

<div class="line"><span class="comment"> * This file is distributed under 3-clause BSD license.</span></div>

<div class="line"><span class="comment"> * The complete license agreement can be obtained at:</span></div>

<div class="line"><span class="comment"> * http://arrayfire.com/licenses/BSD-3-Clause</span></div>

<div class="line"><span class="preprocessor">#include <<a class="code" href="arrayfire_8h.htm">arrayfire.h</a>></span></div>

<div class="line"><span class="preprocessor">#include <iostream></span></div>

<div class="line"><span class="keyword">using namespace </span><a class="code hl_namespace" href="namespaceaf.htm">af</a>;</div>

<div class="line"><span class="keyword">static</span> <span class="keywordtype">size_t</span> dimension = 4 * 1024;</div>

<div class="line"><span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> maxIter = 10;</div>

<div class="line"><span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> sparsityFactor = 7;</div>

<div class="line"><span class="keyword">static</span> <a id="_a0" name="_a0"></a><a class="code hl_class" href="classaf_1_1array.htm">array</a> A;</div>

<div class="line"><span class="keyword">static</span> <a class="code hl_class" href="classaf_1_1array.htm">array</a> spA; <span class="comment">// Sparse A</span></div>

<div class="line"><span class="keyword">static</span> <a class="code hl_class" href="classaf_1_1array.htm">array</a> x0;</div>

<div class="line"><span class="keyword">static</span> <a class="code hl_class" href="classaf_1_1array.htm">array</a> b;</div>

<div class="line"><span class="keywordtype">void</span> setupInputs() {</div>

<div class="line"> <span class="comment">// Generate a random input: A</span></div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> T = randu(dimension, dimension, <a id="a1" name="a1"></a><a class="code hl_enumvalue" href="defines_8h.htm#a023d8ac325fb14f1712a52fb0940b1d5a82ea90203678bdd0b547068f0a76524b">f32</a>);</div>

<div class="line"> <span class="comment">// Create 0s in input.</span></div>

<div class="line"> <span class="comment">// Anything that is no divisible by sparsityFactor will become 0.</span></div>

<div class="line"> A = floor(T * 1000);</div>

<div class="line"> A = A * ((A % sparsityFactor) == 0) / 1000;</div>

<div class="line"> <span class="comment">// Make it positive definite</span></div>

<div class="line"> A = transpose(A) + A + A.<a id="a2" name="a2"></a><a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>(0) * identity(A.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>(0), A.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>(0), <a class="code hl_enumvalue" href="defines_8h.htm#a023d8ac325fb14f1712a52fb0940b1d5a82ea90203678bdd0b547068f0a76524b">f32</a>);</div>

<div class="line"> <span class="comment">// Make A sparse as spA</span></div>

<div class="line"> spA = sparse(A);</div>

<div class="line"> <span class="comment">// Generate x0: Random guess</span></div>

<div class="line"> x0 = randu(A.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>(0), <a class="code hl_enumvalue" href="defines_8h.htm#a023d8ac325fb14f1712a52fb0940b1d5a82ea90203678bdd0b547068f0a76524b">f32</a>);</div>

<div class="line"> <span class="comment">// Generate b</span></div>

<div class="line"> b = matmul(A, x0);</div>

<div class="line"> std::cout << <span class="stringliteral">"Sparsity of A = "</span></div>

<div class="line"> << 100.f * (float)sparseGetNNZ(spA) / (float)spA.<a id="a3" name="a3"></a><a class="code hl_function" href="classaf_1_1array.htm#a89f91cdf327a8b304589916db5bde3c2">elements</a>() << <span class="stringliteral">"%"</span></div>

<div class="line"> std::cout << <span class="stringliteral">"Memory Usage of A = "</span> << A.<a id="a4" name="a4"></a><a class="code hl_function" href="classaf_1_1array.htm#a3b73c6479b9490ad49b8156e14473541">bytes</a>() / (1024.f * 1024.f)</div>

<div class="line"> std::cout << <span class="stringliteral">"Memory Usage of spA = "</span></div>

<div class="line"> << (sparseGetValues(spA).<a class="code hl_function" href="classaf_1_1array.htm#a3b73c6479b9490ad49b8156e14473541">bytes</a>() + sparseGetRowIdx(spA).<a class="code hl_function" href="classaf_1_1array.htm#a3b73c6479b9490ad49b8156e14473541">bytes</a>() +</div>

<div class="line"> sparseGetColIdx(spA).<a class="code hl_function" href="classaf_1_1array.htm#a3b73c6479b9490ad49b8156e14473541">bytes</a>()) /</div>

<div class="line"><span class="keywordtype">void</span> sparseConjugateGradient(<span class="keywordtype">void</span>) {</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> x = constant(0, b.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>(), <a class="code hl_enumvalue" href="defines_8h.htm#a023d8ac325fb14f1712a52fb0940b1d5a82ea90203678bdd0b547068f0a76524b">f32</a>);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> r = b - matmul(spA, x);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> p = r;</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> Ap = matmul(spA, p);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> alpha_num = dot(r, r);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> alpha_den = dot(p, Ap);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> alpha = alpha_num / alpha_den;</div>

<div class="line"> r -= tile(alpha, Ap.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>()) * Ap;</div>

<div class="line"> x += tile(alpha, Ap.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>()) * p;</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> beta_num = dot(r, r);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> beta = beta_num / alpha_num;</div>

<div class="line"><span class="keywordtype">void</span> denseConjugateGradient(<span class="keywordtype">void</span>) {</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> x = <a class="code hl_function" href="group__data__func__constant.htm#ga0db7363474e85542f16e4bbcf57fe842">constant</a>(0, b.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>(), <a class="code hl_enumvalue" href="defines_8h.htm#a023d8ac325fb14f1712a52fb0940b1d5a82ea90203678bdd0b547068f0a76524b">f32</a>);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> r = b - <a class="code hl_function" href="group__blas__func__matmul.htm#gac061af289fcd39a07a3efba0f33fb17f">matmul</a>(A, x);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> p = r;</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> Ap = <a class="code hl_function" href="group__blas__func__matmul.htm#gac061af289fcd39a07a3efba0f33fb17f">matmul</a>(A, p);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> alpha_num = <a class="code hl_function" href="group__blas__func__dot.htm#gab98b9a8de187c5e58e946658e62161ca">dot</a>(r, r);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> alpha_den = <a class="code hl_function" href="group__blas__func__dot.htm#gab98b9a8de187c5e58e946658e62161ca">dot</a>(p, Ap);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> alpha = alpha_num / alpha_den;</div>

<div class="line"> r -= <a class="code hl_function" href="group__manip__func__tile.htm#gaeba95584319d7d90e2c371391db0083b">tile</a>(alpha, Ap.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>()) * Ap;</div>

<div class="line"> x += <a class="code hl_function" href="group__manip__func__tile.htm#gaeba95584319d7d90e2c371391db0083b">tile</a>(alpha, Ap.<a class="code hl_function" href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">dims</a>()) * p;</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> beta_num = <a class="code hl_function" href="group__blas__func__dot.htm#gab98b9a8de187c5e58e946658e62161ca">dot</a>(r, r);</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> beta = beta_num / alpha_num;</div>

<div class="line"><span class="keywordtype">void</span> checkConjugateGradient(<span class="keyword">const</span> <a class="code hl_class" href="classaf_1_1array.htm">af::array</a> in) {</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> p = r;</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> alpha = alpha_num / alpha_den;</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> beta = beta_num / alpha_num;</div>

<div class="line"> <a class="code hl_class" href="classaf_1_1array.htm">array</a> res = x0 - x;</div>

<div class="line"> std::cout << <span class="stringliteral">"Final difference in solutions:\n"</span>;</div>

<div class="line"> <a id="a5" name="a5"></a><a class="code hl_define" href="util_8h.htm#a6de8e7f9eb3914d1cea66e55e7b791d3">af_print</a>(<a class="code hl_function" href="group__blas__func__dot.htm#gab98b9a8de187c5e58e946658e62161ca">dot</a>(res, res));</div>

<div class="line"> setupInputs();</div>

<div class="line"> std::cout << <span class="stringliteral">"Verifying Dense Conjugate Gradient:"</span> << std::endl;</div>

<div class="line"> checkConjugateGradient(A);</div>

<div class="line"> std::cout << <span class="stringliteral">"Verifying Sparse Conjugate Gradient:"</span> << std::endl;</div>

<div class="line"> checkConjugateGradient(spA);</div>

<div class="line"> std::cout << <span class="stringliteral">"Dense Conjugate Gradient Time: "</span></div>

<div class="line"> << <a class="code hl_function" href="namespaceaf.htm#a08955393264b77be242246c0b1148687">timeit</a>(denseConjugateGradient) * 1000 << <span class="stringliteral">"ms"</span> << std::endl;</div>

<div class="line"> std::cout << <span class="stringliteral">"Sparse Conjugate Gradient Time: "</span></div>

<div class="line"> << <a class="code hl_function" href="namespaceaf.htm#a08955393264b77be242246c0b1148687">timeit</a>(sparseConjugateGradient) * 1000 << <span class="stringliteral">"ms"</span> << std::endl;</div>

<div class="line"> <span class="keywordflow">return</span> 0;</div>

<div class="ttc" id="aarrayfire_8h_htm"><div class="ttname"><a href="arrayfire_8h.htm">arrayfire.h</a></div></div>

<div class="ttc" id="aclassaf_1_1array_htm"><div class="ttname"><a href="classaf_1_1array.htm">af::array</a></div><div class="ttdoc">A multi dimensional data container.</div><div class="ttdef"><b>Definition:</b> <a href="array_8h_source.htm#l00037">array.h:37</a></div></div>

<div class="ttc" id="aclassaf_1_1array_htm_a10d14b019693bc98f672ad7c4ac60bac"><div class="ttname"><a href="classaf_1_1array.htm#a10d14b019693bc98f672ad7c4ac60bac">af::array::dims</a></div><div class="ttdeci">dim4 dims() const</div><div class="ttdoc">Get dimensions of the array.</div></div>

<div class="ttc" id="aclassaf_1_1array_htm_a3b73c6479b9490ad49b8156e14473541"><div class="ttname"><a href="classaf_1_1array.htm#a3b73c6479b9490ad49b8156e14473541">af::array::bytes</a></div><div class="ttdeci">size_t bytes() const</div><div class="ttdoc">Get the size of the array in bytes.</div></div>

<div class="ttc" id="aclassaf_1_1array_htm_a89f91cdf327a8b304589916db5bde3c2"><div class="ttname"><a href="classaf_1_1array.htm#a89f91cdf327a8b304589916db5bde3c2">af::array::elements</a></div><div class="ttdeci">dim_t elements() const</div><div class="ttdoc">Get the total number of elements across all dimensions of the array.</div></div>

<div class="ttc" id="adefines_8h_htm_a023d8ac325fb14f1712a52fb0940b1d5a82ea90203678bdd0b547068f0a76524b"><div class="ttname"><a href="defines_8h.htm#a023d8ac325fb14f1712a52fb0940b1d5a82ea90203678bdd0b547068f0a76524b">f32</a></div><div class="ttdeci">@ f32</div><div class="ttdoc">32-bit floating point values</div><div class="ttdef"><b>Definition:</b> <a href="defines_8h_source.htm#l00211">defines.h:211</a></div></div>

<div class="ttc" id="agroup__blas__func__dot_htm_gab98b9a8de187c5e58e946658e62161ca"><div class="ttname"><a href="group__blas__func__dot.htm#gab98b9a8de187c5e58e946658e62161ca">af::dot</a></div><div class="ttdeci">T dot(const array &lhs, const array &rhs, const matProp optLhs=AF_MAT_NONE, const matProp optRhs=AF_MAT_NONE)</div><div class="ttdoc">C++ Interface to compute the dot product.</div></div>

<div class="ttc" id="agroup__blas__func__matmul_htm_gac061af289fcd39a07a3efba0f33fb17f"><div class="ttname"><a href="group__blas__func__matmul.htm#gac061af289fcd39a07a3efba0f33fb17f">af::matmul</a></div><div class="ttdeci">AFAPI array matmul(const array &lhs, const array &rhs, const matProp optLhs=AF_MAT_NONE, const matProp optRhs=AF_MAT_NONE)</div><div class="ttdoc">C++ Interface to multiply two matrices.</div></div>

<div class="ttc" id="agroup__data__func__constant_htm_ga0db7363474e85542f16e4bbcf57fe842"><div class="ttname"><a href="group__data__func__constant.htm#ga0db7363474e85542f16e4bbcf57fe842">af::constant</a></div><div class="ttdeci">array constant(T val, const dim4 &dims, const dtype ty=(af_dtype) dtype_traits< T >::ctype)</div><div class="ttdoc">C++ Interface to generate an array with elements set to a specified value.</div></div>

<div class="ttc" id="agroup__device__func__info_htm_ga3e22b6ff6e0940bb31577bea2863ca27"><div class="ttname"><a href="group__device__func__info.htm#ga3e22b6ff6e0940bb31577bea2863ca27">af::info</a></div><div class="ttdeci">AFAPI void info()</div></div>

<div class="ttc" id="agroup__device__func__sync_htm_ga4e9c28bad482dc590cac930515f14bfb"><div class="ttname"><a href="group__device__func__sync.htm#ga4e9c28bad482dc590cac930515f14bfb">af::sync</a></div><div class="ttdeci">AFAPI void sync(const int device=-1)</div><div class="ttdoc">Blocks until the device is finished processing.</div></div>

<div class="ttc" id="agroup__manip__func__tile_htm_gaeba95584319d7d90e2c371391db0083b"><div class="ttname"><a href="group__manip__func__tile.htm#gaeba95584319d7d90e2c371391db0083b">af::tile</a></div><div class="ttdeci">AFAPI array tile(const array &in, const unsigned x, const unsigned y=1, const unsigned z=1, const unsigned w=1)</div><div class="ttdoc">C++ Interface to generate a tiled array.</div></div>

<div class="ttc" id="anamespaceaf_htm"><div class="ttname"><a href="namespaceaf.htm">af</a></div><div class="ttdef"><b>Definition:</b> <a href="algorithm_8h_source.htm#l00014">algorithm.h:15</a></div></div>

<div class="ttc" id="anamespaceaf_htm_a08955393264b77be242246c0b1148687"><div class="ttname"><a href="namespaceaf.htm#a08955393264b77be242246c0b1148687">af::timeit</a></div><div class="ttdeci">AFAPI double timeit(void(*fn)())</div></div>

<div class="ttc" id="autil_8h_htm_a6de8e7f9eb3914d1cea66e55e7b791d3"><div class="ttname"><a href="util_8h.htm#a6de8e7f9eb3914d1cea66e55e7b791d3">af_print</a></div><div class="ttdeci">#define af_print(...)</div><div class="ttdef"><b>Definition:</b> <a href="util_8h_source.htm#l00148">util.h:148</a></div></div>

</div> </div>

</div>

<ul>

<li class="footer">Generated by <a href="https://www.doxygen.org/index.html"><img class="footer" src="doxygen.svg" width="104" height="31" alt="doxygen"/></a> 1.9.5 </li>

</ul>

</div>

</body>

</html>

arrayfire.github.io/docs/benchmarks_2cg_8cpp-example.htm at master · arrayfire/arrayfire.github.io · GitHub

Uh oh!

Expand file tree

Breadcrumbs

benchmarks_2cg_8cpp-example.htm

Latest commit

History

Breadcrumbs

benchmarks_2cg_8cpp-example.htm

File metadata and controls