Skip to content

Commit 9242f8b

Browse files
authored
Fix SIMD ifdefs in ngpFieldOps (#1322)
1 parent a75e794 commit 9242f8b

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

include/ngp_utils/NgpFieldOps.h

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ struct NodeFieldOp
151151
const auto& msh = obj_.ngpMesh_;
152152
const auto& fld = obj_.ngpField_;
153153
const auto* einfo = edata_.info();
154-
#ifdef STK_SIMD_NONE
154+
#ifdef USE_STK_SIMD_NONE
155155
fld.get(msh, einfo[0].entityNodes[ni], ic) = stk::simd::get_data(val, 0);
156156
#else
157157
for (int is = 0; is < edata_.numSimdElems; ++is) {
@@ -167,7 +167,7 @@ struct NodeFieldOp
167167
const auto& msh = obj_.ngpMesh_;
168168
const auto& fld = obj_.ngpField_;
169169
const auto* einfo = edata_.info();
170-
#ifdef STK_SIMD_NONE
170+
#ifdef USE_STK_SIMD_NONE
171171
Kokkos::atomic_add(
172172
&fld.get(msh, einfo[0].entityNodes[ni], ic),
173173
stk::simd::get_data(val, 0));
@@ -186,7 +186,7 @@ struct NodeFieldOp
186186
const auto& msh = obj_.ngpMesh_;
187187
const auto& fld = obj_.ngpField_;
188188
const auto* einfo = edata_.info();
189-
#ifdef STK_SIMD_NONE
189+
#ifdef USE_STK_SIMD_NONE
190190
fld.get(msh, einfo[0].entityNodes[ni], ic) = val;
191191
#else
192192
for (int is = 0; is < edata_.numSimdElems; ++is) {
@@ -201,7 +201,7 @@ struct NodeFieldOp
201201
const auto& msh = obj_.ngpMesh_;
202202
const auto& fld = obj_.ngpField_;
203203
const auto* einfo = edata_.info();
204-
#ifdef STK_SIMD_NONE
204+
#ifdef USE_STK_SIMD_NONE
205205
Kokkos::atomic_add(&fld.get(msh, einfo[0].entityNodes[ni], ic), val);
206206
#else
207207
for (int is = 0; is < edata_.numSimdElems; ++is) {
@@ -274,7 +274,7 @@ struct ElemFieldOp
274274
{
275275
const auto& fld = obj_.ngpField_;
276276
const auto* einfo = edata_.info();
277-
#ifdef STK_SIMD_NONE
277+
#ifdef USE_STK_SIMD_NONE
278278
fld.get(einfo[0].meshIdx, ic) = stk::simd::get_data(val, 0);
279279
#else
280280
for (int is = 0; is < edata_.numSimdElems; ++is) {
@@ -290,7 +290,7 @@ struct ElemFieldOp
290290
const auto* einfo = edata_.info();
291291

292292
// No atomic_add here as only one element active per thread
293-
#ifdef STK_SIMD_NONE
293+
#ifdef USE_STK_SIMD_NONE
294294
fld.get(einfo[0].meshIdx, ic) += stk::simd::get_data(val, 0);
295295
#else
296296
for (int is = 0; is < edata_.numSimdElems; ++is) {
@@ -304,7 +304,7 @@ struct ElemFieldOp
304304
{
305305
const auto& fld = obj_.ngpField_;
306306
const auto* einfo = edata_.info();
307-
#ifdef STK_SIMD_NONE
307+
#ifdef USE_STK_SIMD_NONE
308308
fld.get(einfo[0].meshIdx, ic) = val;
309309
#else
310310
for (int is = 0; is < edata_.numSimdElems; ++is) {
@@ -318,7 +318,7 @@ struct ElemFieldOp
318318
{
319319
const auto& fld = obj_.ngpField_;
320320
const auto* einfo = edata_.info();
321-
#ifdef STK_SIMD_NONE
321+
#ifdef USE_STK_SIMD_NONE
322322
fld.get(einfo[0].meshIdx, ic) += val;
323323
#else
324324
for (int is = 0; is < edata_.numSimdElems; ++is) {

0 commit comments

Comments
 (0)