Skip to content

Commit 53f6316

Browse files
committed
changed access function in Field class to operator()
1 parent df7547e commit 53f6316

File tree

2 files changed

+44
-44
lines changed

2 files changed

+44
-44
lines changed

src/MeshField.hpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,22 +16,22 @@ class Field {
1616
}
1717

1818
KOKKOS_INLINE_FUNCTION
19-
auto& access(int s, int a) const {
19+
auto& operator()(int s, int a) const {
2020
return slice.access(s,a);
2121
}
2222

2323
KOKKOS_INLINE_FUNCTION
24-
auto& access(int s, int a, int i) const {
24+
auto& operator()(int s, int a, int i) const {
2525
return slice.access(s,a,i);
2626
}
2727

2828
KOKKOS_INLINE_FUNCTION
29-
auto& access(int s, int a, int i, int j) const {
29+
auto& operator()(int s, int a, int i, int j) const {
3030
return slice.access(s,a,i,j);
3131
}
3232

3333
KOKKOS_INLINE_FUNCTION
34-
auto& access(int s, int a, int i, int j, int k) const {
34+
auto& operator()(int s, int a, int i, int j, int k) const {
3535
return slice.access(s,a,i,j,k);
3636
}
3737
};

test/MeshField.cpp

Lines changed: 40 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ void single_type(int num_tuples) {
1919
auto vector_kernel = KOKKOS_LAMBDA(const int s, const int a)
2020
{
2121
double d0 = 10;
22-
field0.access(s,a) = d0;
23-
assert(field0.access(s,a) == d0);
22+
field0(s,a) = d0;
23+
assert(field0(s,a) == d0);
2424
};
2525

2626
cabMeshField.parallel_for(0,num_tuples,vector_kernel,"single_type_pfor");
@@ -44,21 +44,21 @@ void multi_type(int num_tuples) {
4444
auto vector_kernel = KOKKOS_LAMBDA(const int s, const int a)
4545
{
4646
double d0 = 10.456;
47-
field0.access(s,a) = d0;
47+
field0(s,a) = d0;
4848
double d1 = 43.973234567;
49-
field1.access(s,a) = d1;
49+
field1(s,a) = d1;
5050
float f0 = 123.45;
51-
field2.access(s,a) = f0;
51+
field2(s,a) = f0;
5252
int i0 = 22;
53-
field3.access(s,a) = i0;
53+
field3(s,a) = i0;
5454
char c0 = 'a';
55-
field4.access(s,a) = c0;
55+
field4(s,a) = c0;
5656

57-
assert(field0.access(s,a) == d0);
58-
assert(field1.access(s,a) == d1);
59-
assert(field2.access(s,a) == f0);
60-
assert(field3.access(s,a) == i0);
61-
assert(field4.access(s,a) == c0);
57+
assert(field0(s,a) == d0);
58+
assert(field1(s,a) == d1);
59+
assert(field2(s,a) == f0);
60+
assert(field3(s,a) == i0);
61+
assert(field4(s,a) == c0);
6262
};
6363

6464
cabMeshField.parallel_for(0,num_tuples,vector_kernel,"multi_type_pfor");
@@ -85,27 +85,27 @@ void many_type(int num_tuples) {
8585
auto vector_kernel = KOKKOS_LAMBDA(const int s, const int a)
8686
{
8787
double d0 = 10.456;
88-
field0.access(s,a) = d0;
88+
field0(s,a) = d0;
8989
double d1 = 43.973234567;
90-
field1.access(s,a) = d1;
90+
field1(s,a) = d1;
9191
float f0 = 123.45;
92-
field2.access(s,a) = f0;
92+
field2(s,a) = f0;
9393
float f1 = 543.21;
94-
field3.access(s,a) = f1;
94+
field3(s,a) = f1;
9595
int i0 = 222;
96-
field4.access(s,a) = i0;
96+
field4(s,a) = i0;
9797
short int i1 = 50;
98-
field5.access(s,a) = i1;
98+
field5(s,a) = i1;
9999
char c0 = 'h';
100-
field6.access(s,a) = c0;
100+
field6(s,a) = c0;
101101

102-
assert(field0.access(s,a) == d0);
103-
assert(field1.access(s,a) == d1);
104-
assert(field2.access(s,a) == f0);
105-
assert(field3.access(s,a) == f1);
106-
assert(field4.access(s,a) == i0);
107-
assert(field5.access(s,a) == i1);
108-
assert(field6.access(s,a) == c0);
102+
assert(field0(s,a) == d0);
103+
assert(field1(s,a) == d1);
104+
assert(field2(s,a) == f0);
105+
assert(field3(s,a) == f1);
106+
assert(field4(s,a) == i0);
107+
assert(field5(s,a) == i1);
108+
assert(field6(s,a) == c0);
109109
};
110110

111111
cabMeshField.parallel_for(0,num_tuples,vector_kernel,"many_type_pfor");
@@ -127,8 +127,8 @@ void rank1_arr(int num_tuples) {
127127
for (int i = 0; i < width; i++)
128128
{
129129
double d0 = 10+i;
130-
field0.access(s,a,i) = d0;
131-
assert(field0.access(s,a,i) == d0);
130+
field0(s,a,i) = d0;
131+
assert(field0(s,a,i) == d0);
132132
}
133133
};
134134

@@ -154,8 +154,8 @@ void rank2_arr(int num_tuples) {
154154
for (int j = 0; j < height; j++)
155155
{
156156
double d0 = (10+i)/(j+1);
157-
field0.access(s,a,i,j) = d0;
158-
assert(field0.access(s,a,i,j) == d0);
157+
field0(s,a,i,j) = d0;
158+
assert(field0(s,a,i,j) == d0);
159159
}
160160
}
161161
};
@@ -185,8 +185,8 @@ void rank3_arr(int num_tuples) {
185185
for (int k = 0; k < depth; k++)
186186
{
187187
double d0 = ((10+i)*(k+1))/(j+1);
188-
field0.access(s,a,i,j,k) = d0;
189-
assert(field0.access(s,a,i,j,k) == d0);
188+
field0(s,a,i,j,k) = d0;
189+
assert(field0(s,a,i,j,k) == d0);
190190
}
191191
}
192192
}
@@ -219,27 +219,27 @@ void mix_arr(int num_tuples) {
219219
float f0;
220220
int i0;
221221
char c0 = 's';
222-
field3.access(s,a) = c0;
222+
field3(s,a) = c0;
223223

224224
for (int i = 0; i < width; i++)
225225
{
226226
i0 = i+s+a;
227-
field2.access(s,a,i) = i0;
227+
field2(s,a,i) = i0;
228228
for (int j = 0; j < height; j++)
229229
{
230230
f0 = i0 / (i+j+1.123);
231-
field1.access(s,a,i,j) = f0;
231+
field1(s,a,i,j) = f0;
232232
for (int k = 0; k < depth; k++)
233233
{
234234
double d0 = ((10+i)*(k+1))/(j+1);
235-
field0.access(s,a,i,j,k) = d0;
236-
assert(field0.access(s,a,i,j,k) == d0);
235+
field0(s,a,i,j,k) = d0;
236+
assert(field0(s,a,i,j,k) == d0);
237237
}
238-
assert(field1.access(s,a,i,j) == f0);
238+
assert(field1(s,a,i,j) == f0);
239239
}
240-
assert(field2.access(s,a,i) == i0);
240+
assert(field2(s,a,i) == i0);
241241
}
242-
assert(field3.access(s,a) == c0);
242+
assert(field3(s,a) == c0);
243243
};
244244

245245
cabMeshField.parallel_for(0,num_tuples,vector_kernel,"mix_arr_pfor");

0 commit comments

Comments
 (0)