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
|
/*
-- MAGMA (version 2.9.0) --
Univ. of Tennessee, Knoxville
Univ. of California, Berkeley
Univ. of Colorado, Denver
@date January 2025
@precisions normal z -> s d c
@author Hartwig Anzt
*/
#include "magmasparse_internal.h"
#define THRESHOLD 10e-99
/**
Purpose
-------
Computes the Frobenius norm of the difference between the CSR matrices A
and B. They do not need to share the same sparsity pattern!
res = ||A-B||_F = sqrt( sum_ij (A_ij-B_ij)^2 )
Arguments
---------
@param[in]
A magma_z_matrix
sparse matrix in CSR
@param[in]
B magma_z_matrix
sparse matrix in CSR
@param[out]
res real_Double_t*
residual
@param[in]
queue magma_queue_t
Queue to execute in.
@ingroup magmasparse_zaux
********************************************************************/
extern "C" magma_int_t
magma_zmdiff(
magma_z_matrix A, magma_z_matrix B,
real_Double_t *res,
magma_queue_t queue )
{
magma_int_t info = 0;
if ( A.memory_location == Magma_CPU && B.memory_location == Magma_CPU
&& A.storage_type == Magma_CSR && B.storage_type == Magma_CSR ){
real_Double_t tmp2;
magma_int_t i,j,k;
*res = 0.0;
for(i=0; i<A.num_rows; i++) {
for(j=A.row[i]; j<A.row[i+1]; j++) {
magma_index_t localcol = A.col[j];
for( k=B.row[i]; k<B.row[i+1]; k++) {
if (B.col[k] == localcol) {
tmp2 = (real_Double_t) fabs( MAGMA_Z_REAL(A.val[j] )
- MAGMA_Z_REAL(B.val[k]) );
(*res) = (*res) + tmp2* tmp2;
}
}
}
}
(*res) = sqrt((*res));
}
else {
printf("error: mdiff only supported for CSR matrices on the CPU: %d %d %d %d.\n",
int(A.memory_location), int(B.memory_location), int(A.storage_type), int(B.storage_type));
info = MAGMA_ERR_NOT_SUPPORTED;
}
return info;
}
|
