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
|
/*
-- 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 "testings.h"
/* ////////////////////////////////////////////////////////////////////////////
-- testing csr matrix add
*/
int main( int argc, char** argv )
{
magma_int_t info = 0;
TESTING_CHECK( magma_init() );
magma_print_environment();
magma_queue_t queue=NULL;
magma_queue_create( 0, &queue );
real_Double_t res;
magma_z_matrix A={Magma_CSR}, B={Magma_CSR}, B2={Magma_CSR},
dA={Magma_CSR}, dB={Magma_CSR}, dC={Magma_CSR};
magmaDoubleComplex one = MAGMA_Z_MAKE(1.0, 0.0);
magmaDoubleComplex mone = MAGMA_Z_MAKE(-1.0, 0.0);
magma_int_t i=1;
if ( strcmp("LAPLACE2D", argv[i]) == 0 && i+1 < argc ) { // Laplace test
i++;
magma_int_t laplace_size = atoi( argv[i] );
TESTING_CHECK( magma_zm_5stencil( laplace_size, &A, queue ));
} else { // file-matrix test
TESTING_CHECK( magma_z_csr_mtx( &A, argv[i], queue ));
}
printf("%% matrix info: %lld-by-%lld with %lld nonzeros\n",
(long long) A.num_rows, (long long) A.num_cols, (long long) A.nnz );
i++;
if ( strcmp("LAPLACE2D", argv[i]) == 0 && i+1 < argc ) { // Laplace test
i++;
magma_int_t laplace_size = atoi( argv[i] );
TESTING_CHECK( magma_zm_5stencil( laplace_size, &B, queue ));
} else { // file-matrix test
TESTING_CHECK( magma_z_csr_mtx( &B, argv[i], queue ));
}
printf("%% matrix info: %lld-by-%lld with %lld nonzeros\n",
(long long) B.num_rows, (long long) B.num_cols, (long long) B.nnz );
TESTING_CHECK( magma_zmtransfer( A, &dA, Magma_CPU, Magma_DEV, queue ));
TESTING_CHECK( magma_zmtransfer( B, &dB, Magma_CPU, Magma_DEV, queue ));
TESTING_CHECK( magma_zcuspaxpy( &one, dA, &one, dB, &dC, queue ));
magma_zmfree(&dB, queue );
TESTING_CHECK( magma_zcuspaxpy( &mone, dA, &one, dC, &dB, queue ));
TESTING_CHECK( magma_zmtransfer( dB, &B2, Magma_DEV, Magma_CPU, queue ));
magma_zmfree(&dA, queue );
magma_zmfree(&dB, queue );
magma_zmfree(&dC, queue );
// check difference
TESTING_CHECK( magma_zmdiff( B, B2, &res, queue ));
printf("%% ||A-B||_F = %8.2e\n", res);
if ( res < .000001 )
printf("%% tester matrix add: ok\n");
else
printf("%% tester matrix add: failed\n");
magma_zmfree(&A, queue );
magma_zmfree(&B, queue );
magma_zmfree(&B2, queue );
magma_queue_destroy( queue );
TESTING_CHECK( magma_finalize() );
return info;
}
|
