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
|
/*
-- MAGMA (version 2.9.0) --
Univ. of Tennessee, Knoxville
Univ. of California, Berkeley
Univ. of Colorado, Denver
@date January 2025
@generated from sparse/control/magma_zmfrobenius.cpp, normal z -> d, Wed Jan 22 14:42:32 2025
@author Hartwig Anzt
*/
#include "magmasparse_internal.h"
#define PRECISION_d
/**
Purpose
-------
Computes the Frobenius norm || A - B ||_S on the sparsity pattern of S.
Arguments
---------
@param[in]
A magma_d_matrix
input sparse matrix in CSR
@param[in]
B magma_d_matrix
input sparse matrix in CSR
@param[in]
S magma_d_matrix
input sparsity pattern in CSR
@param[out]
norm double*
Frobenius norm of difference on sparsity pattern S
@param[in]
queue magma_queue_t
Queue to execute in.
@ingroup magmasparse_daux
********************************************************************/
extern "C" magma_int_t
magma_dmfrobenius(
magma_d_matrix A,
magma_d_matrix B,
magma_d_matrix S,
double *norm,
magma_queue_t queue )
{
magma_int_t info = 0;
double tmp;
magma_int_t i,j,k;
magma_d_matrix hA={Magma_CSR}, hB={Magma_CSR}, hS={Magma_CSR};
CHECK( magma_dmtransfer( A, &hA, A.memory_location, Magma_CPU, queue ));
CHECK( magma_dmtransfer( B, &hB, B.memory_location, Magma_CPU, queue ));
CHECK( magma_dmtransfer( S, &hS, S.memory_location, Magma_CPU, queue ));
if( hA.num_rows == hB.num_rows && hA.num_rows == hS.num_rows ) {
for(i=0; i<hS.num_rows; i++){
for(j=hS.row[i]; j<hS.row[i+1]; j++){
magma_index_t lcol = hS.col[j];
double Aval = MAGMA_D_MAKE(0.0, 0.0);
double Bval = MAGMA_D_MAKE(0.0, 0.0);
for(k=hA.row[i]; k<hA.row[i+1]; k++){
if( hA.col[k] == lcol ){
Aval = hA.val[k];
}
}
for(k=hB.row[i]; k<hB.row[i+1]; k++){
if( hB.col[k] == lcol ){
Bval = hB.val[k];
}
}
tmp = MAGMA_D_ABS(Aval - Bval) ;
(*norm) = (*norm) + tmp * tmp;
}
}
(*norm) = sqrt((*norm));
}
cleanup:
magma_dmfree( &hA, queue );
magma_dmfree( &hB, queue );
magma_dmfree( &hS, queue );
return info;
}
|
