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
|
/*
-- 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_zmcsrcompressor.cpp, normal z -> s, Wed Jan 22 14:42:27 2025
@author Hartwig Anzt
*/
#include "magmasparse_internal.h"
/**
Purpose
-------
Removes zeros in a CSR matrix.
Arguments
---------
@param[in,out]
A magma_s_matrix*
input/output matrix
@param[in]
queue magma_queue_t
Queue to execute in.
@ingroup magmasparse_saux
********************************************************************/
extern "C" magma_int_t
magma_smcsrcompressor(
magma_s_matrix *A,
magma_queue_t queue )
{
magma_int_t info = 0;
magma_s_matrix B={Magma_CSR};
magma_s_matrix hA={Magma_CSR}, CSRA={Magma_CSR};
if (A->ownership) {
if ( A->memory_location == Magma_CPU && A->storage_type == Magma_CSR ) {
CHECK( magma_smconvert( *A, &B, Magma_CSR, Magma_CSR, queue ));
magma_free_cpu( A->row );
magma_free_cpu( A->col );
magma_free_cpu( A->val );
CHECK( magma_s_csr_compressor(&B.val, &B.row, &B.col,
&A->val, &A->row, &A->col, &A->num_rows, queue ));
A->nnz = A->row[A->num_rows];
}
else {
magma_storage_t A_storage = A->storage_type;
magma_location_t A_location = A->memory_location;
CHECK( magma_smtransfer( *A, &hA, A->memory_location, Magma_CPU, queue ));
CHECK( magma_smconvert( hA, &CSRA, hA.storage_type, Magma_CSR, queue ));
CHECK( magma_smcsrcompressor( &CSRA, queue ));
magma_smfree( &hA, queue );
magma_smfree( A, queue );
CHECK( magma_smconvert( CSRA, &hA, Magma_CSR, A_storage, queue ));
CHECK( magma_smtransfer( hA, A, Magma_CPU, A_location, queue ));
magma_smfree( &hA, queue );
magma_smfree( &CSRA, queue );
}
}
cleanup:
magma_smfree( &hA, queue );
magma_smfree( &CSRA, queue );
magma_smfree( &B, queue );
return info;
}
|
