1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
// Copyright (c) 2017-2023, University of Tennessee. All rights reserved.
// SPDX-License-Identifier: BSD-3-Clause
// This program is free software: you can redistribute it and/or modify it under
// the terms of the BSD 3-Clause license. See the accompanying LICENSE file.
// blas01_util.cc
// BLAS++ utilities: blas::real_type, blas::is_complex, blas::conj
#include <blas.hh>
#include "util.hh"
//------------------------------------------------------------------------------
template <typename scalar_type>
void test_util( scalar_type alpha )
{
print_func();
//--------------------
// demo blas::real_type
int64_t n=100;
std::vector<scalar_type> x( n, 1.0 );
using real_type = blas::real_type< scalar_type >;
real_type norm = blas::nrm2( n, x.data(), 1 );
printf( "norm %7.4f\n", norm );
//--------------------
// demo blas::conj
scalar_type beta;
// std::conj fails if alpha is real:
// error: cannot convert 'std::complex<double>' to 'double' in assignment
//beta = std::conj( alpha );
// blas::conj works. Need `using`!
using blas::conj;
beta = conj( alpha );
//--------------------
// demo blas::is_complex
using std::real;
using std::imag;
if (blas::is_complex_v<scalar_type>) {
printf( "alpha %7.4f + %7.4fi\n", real(alpha), imag(alpha) );
printf( "beta %7.4f + %7.4fi\n", real(beta), imag(beta) );
}
else {
printf( "alpha %7.4f\n", real(alpha) );
printf( "beta %7.4f\n", real(beta) );
}
}
//------------------------------------------------------------------------------
int main( int argc, char** argv )
{
try {
// Parse command line to set types for s, d, c, z precisions.
bool types[ 4 ];
parse_args( argc, argv, types );
// Run tests.
int m = 100, n = 200, k = 50;
if (types[ 0 ])
test_util( float(1.234) );
if (types[ 1 ])
test_util( double(2.468) );
if (types[ 2 ])
test_util( std::complex< float>( 3.1415, 0.5678 ) );
if (types[ 3 ])
test_util( std::complex<double>( 6.2830, 1.1356 ) );
}
catch (std::exception const& ex) {
fprintf( stderr, "%s", ex.what() );
return 1;
}
return 0;
}
|
