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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
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
|
/*
-- MAGMA (version 2.9.0) --
Univ. of Tennessee, Knoxville
Univ. of California, Berkeley
Univ. of Colorado, Denver
@date January 2025
@precisions normal z -> c d s
@author Hartwig Anzt
*/
// includes, system
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <math.h>
// includes, project
#include "magma_v2.h"
#include "magmasparse.h"
#include "magma_operators.h"
#include "testings.h"
/* ////////////////////////////////////////////////////////////////////////////
-- testing zdot
*/
int main( int argc, char** argv )
{
magma_int_t info = 0;
magma_queue_t queue=NULL;
magma_queue_create( 0, &queue );
const magmaDoubleComplex one = MAGMA_Z_MAKE(1.0, 0.0);
const magmaDoubleComplex zero = MAGMA_Z_MAKE(0.0, 0.0);
magmaDoubleComplex alpha;
TESTING_CHECK( magma_init() );
magma_print_environment();
magma_z_matrix a={Magma_CSR}, b={Magma_CSR}, x={Magma_CSR}, y={Magma_CSR}, skp={Magma_CSR};
printf("%%================================================================================================================================================\n");
printf("\n");
printf("%% | runtime | GFLOPS\n");
printf("%% n num_vecs | CUDOT MAGMA MDOTC | CUDOT MAGMA MDOTC\n");
printf("%%------------------------------------------------------------------------------------------------------------------------------------------------\n");
printf("\n");
for( magma_int_t num_vecs=2; num_vecs < 9; num_vecs += 2 ) {
for( magma_int_t n=1000000; n < 5000001; n += 1000000 ) {
int iters = 10;
double computations = ( n * iters * num_vecs);
#ifndef ENABLE_TIMER
#define ENABLE_TIMER
#endif
#ifdef ENABLE_TIMER
real_Double_t mdot1, mdot2, cudot1, cudot2;
real_Double_t mdot_time, cudot_time;
#endif
TESTING_CHECK( magma_zvinit( &a, Magma_DEV, n, num_vecs, one, queue ));
TESTING_CHECK( magma_zvinit( &b, Magma_DEV, num_vecs, 1, one, queue ));
int min_ten = min(num_vecs, 15);
TESTING_CHECK( magma_zvinit( &x, Magma_DEV, min_ten, n, one, queue ));
TESTING_CHECK( magma_zvinit( &y, Magma_DEV, min_ten, n, one, queue ));
TESTING_CHECK( magma_zvinit( &skp, Magma_DEV, num_vecs, 1, zero, queue ));
// warm up
TESTING_CHECK( magma_zgemvmdot( n, num_vecs, a.dval, b.dval, x.dval, y.dval, skp.dval, queue ));
// CUDOT
#ifdef ENABLE_TIMER
cudot1 = magma_sync_wtime( queue );
#endif
for( int h=0; h < iters; h++) {
for( int l=0; l < num_vecs/2; l++) {
alpha = magma_zdotc( n, a.dval, 1, b.dval, 1, queue );
}
}
#ifdef ENABLE_TIMER
cudot2 = magma_sync_wtime( queue );
cudot_time=cudot2-cudot1;
#endif
// MAGMA MDOTC
#ifdef ENABLE_TIMER
mdot1 = magma_sync_wtime( queue );
#endif
for( int h=0; h < iters; h++) {
if( num_vecs == 2 ){
magma_zmdotc1( n, a.dval, b.dval, x.dval, y.dval, skp.dval, queue );
}
else if( num_vecs == 4 ){
magma_zmdotc1( n, a.dval, b.dval, x.dval, y.dval, skp.dval, queue );
magma_zmdotc1( n, a.dval, b.dval, x.dval, y.dval, skp.dval, queue );
}
else if( num_vecs == 6 ){
magma_zmdotc3( n, a.dval, b.dval, a.dval+n, b.dval+n, a.dval+2*n, b.dval+2*n, x.dval, y.dval, skp.dval, queue );
}
else if( num_vecs == 8 ){
magma_zmdotc3( n, a.dval, b.dval, a.dval+n, b.dval+n, a.dval+2*n, b.dval+2*n, x.dval, y.dval, skp.dval, queue );
}
else{
printf("error: not supported.\n");
}
}
#ifdef ENABLE_TIMER
mdot2 = magma_sync_wtime( queue );
mdot_time=mdot2-mdot1;
#endif
//Chronometry
#ifdef ENABLE_TIMER
printf("%lld %lld %e %e %e %e\n",
(long long) n, (long long) num_vecs,
cudot_time/iters,
(mdot_time)/iters,
computations/(cudot_time*1e9),
computations/(mdot_time*1e9));
#endif
magma_zmfree(&a, queue );
magma_zmfree(&b, queue );
magma_zmfree(&x, queue );
magma_zmfree(&y, queue );
magma_zmfree(&skp, queue );
}
printf("%%================================================================================================================================================\n");
printf("\n");
printf("\n");
}
// use alpha to silence compiler warnings
if ( magma_d_isnan( (double) real( alpha ))) {
info = -1;
}
magma_queue_destroy( queue );
TESTING_CHECK( magma_finalize() );
return info;
}
|
