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
|
/*
-- 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_zdomainoverlap.cpp, normal z -> c, Wed Jan 22 14:42:24 2025
@author Hartwig Anzt
*/
#include "magmasparse_internal.h"
extern "C"
magma_int_t
magma_cindexcopy(
magma_int_t num_copy,
magma_int_t offset,
magma_index_t *tmp_x,
magma_index_t *x,
magma_queue_t queue )
{
magma_int_t info = 0;
CHECK( magma_cindexsort( tmp_x, 0, num_copy-1, queue ));
for( magma_int_t j=0; j<num_copy; j++ ){
x[ j+offset ] = tmp_x[ j ];
tmp_x[ j ] = -1;
}
cleanup:
return info;
}
/**
Purpose
-------
Generates the update list.
Arguments
---------
@param[in]
x magma_index_t*
array to sort
@param[in]
num_rows magma_int_t
number of rows in matrix
@param[out]
num_indices magma_int_t*
number of indices in array
@param[in]
rowptr magma_index_t*
rowpointer of matrix
@param[in]
colidx magma_index_t*
colindices of matrix
@param[in]
x magma_index_t*
array containing indices for domain overlap
@param[in]
queue magma_queue_t
Queue to execute in.
@ingroup magmasparse_caux
********************************************************************/
extern "C"
magma_int_t
magma_cdomainoverlap(
magma_index_t num_rows,
magma_int_t *num_indices,
magma_index_t *rowptr,
magma_index_t *colidx,
magma_index_t *x,
magma_queue_t queue )
{
magma_int_t info = 0;
magma_int_t blocksize=128;
magma_int_t row=0, col=0, num_ind=0, offset=0;
magma_index_t *tmp_x;
CHECK( magma_index_malloc_cpu( &tmp_x, blocksize ));
for(magma_int_t i=0; i<blocksize; i++ ){
tmp_x[i] = -1;
}
for(magma_int_t i=0; i<num_rows; i++){
row = i;
for(magma_int_t j=rowptr[row]; j<rowptr[row+1]; j++){
col = colidx[j];
int floatitem = 0;
for(magma_int_t k=0; k<blocksize; k++){
if (tmp_x[k] == col)
floatitem = 1;
}
if( floatitem == 0 ){
tmp_x[num_ind] = col;
num_ind++;
(*num_indices)++;
}
if( num_ind == blocksize || j == rowptr[num_rows]-1 ){
magma_cindexcopy( num_ind, offset, tmp_x, x, queue );
offset=offset+num_ind;
num_ind = 0;
break;
}
}
}
cleanup:
magma_free_cpu( tmp_x );
return info;
}
|
