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
|
// 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.
#include <stdio.h>
#include <complex>
// Use C99 _Complex as return type to be compatible with extern C linkage.
#include <complex.h>
#include "config.h"
//------------------------------------------------------------------------------
#define BLAS_zdotc FORTRAN_NAME( zdotc, ZDOTC )
// result return directly
#ifdef __cplusplus
extern "C"
#endif
double _Complex BLAS_zdotc(
const blas_int* n,
const std::complex<double>* x, const blas_int* incx,
const std::complex<double>* y, const blas_int* incy );
//------------------------------------------------------------------------------
int main()
{
blas_int n = 5, ione = 1;
std::complex<double> x[] = { 1, 2, 3, 4, 5 };
std::complex<double> y[] = { 5, 4, 3, 2, 1 };
for (int i = 0; i < n; ++i) {
printf( "x[ %d ] = %.1f + %.1fi; y[ %d ] = %.1f + %.1fi\n",
i, real( x[ i ] ), imag( x[ i ] ),
i, real( y[ i ] ), imag( y[ i ] ) );
}
double _Complex r = BLAS_zdotc( &n, x, &ione, y, &ione );
std::complex<double> result = *reinterpret_cast< std::complex<double>* >( &r );
printf( "result = %.1f + %.1fi; should be 35.0 + 0.0i\n",
real( result ), imag( result ) );
bool okay = (real(result) == 35);
printf( "%s\n", okay ? "ok" : "failed" );
return ! okay;
}
|
