@@ -317,7 +317,10 @@ void nrncore2nrn_send_values(NrnThread* nth) {
317
317
double * gather_i = tr->gather [i];
318
318
nrn_pragma_acc (update self (gather_i [0 :1 ]) if (nth->compute_gpu )
319
319
async (nth->streams [nth->stream_id ]))
320
- nrn_pragma_omp (target update from (gather_i [0 :1 ]) if (nth->compute_gpu ) depend (inout: nth->streams [nth->stream_id ]) nowait)
320
+ // clang-format off
321
+ nrn_pragma_omp (target update from (gather_i [0 :1 ]) if (nth->compute_gpu )
322
+ depend (inout: nth->streams [nth->stream_id ]) nowait)
323
+ // clang-format on
321
324
}
322
325
nrn_pragma_acc (wait async (nth->streams [nth->stream_id ]))
323
326
for (int i = 0 ; i < tr->n_trajec ; ++i) {
@@ -341,7 +344,8 @@ static void* nrn_fixed_step_thread(NrnThread* nth) {
341
344
if (nth->ncell ) {
342
345
/* @todo: do we need to update nth->_t on GPU: Yes (Michael, but can
343
346
launch kernel) */
344
- nrn_pragma_acc (update device (nth->_t ) if (nth->compute_gpu ) async (nth->streams [nth->stream_id ]))
347
+ nrn_pragma_acc (update device (nth->_t ) if (nth->compute_gpu )
348
+ async (nth->streams [nth->stream_id ]))
345
349
nrn_pragma_acc (wait async (nth->streams [nth->stream_id ]))
346
350
nrn_pragma_omp (target update to (nth->_t ) if (nth->compute_gpu ))
347
351
fixed_play_continuous (nth);
@@ -377,7 +381,8 @@ void* nrn_fixed_step_lastpart(NrnThread* nth) {
377
381
378
382
if (nth->ncell ) {
379
383
/* @todo: do we need to update nth->_t on GPU */
380
- nrn_pragma_acc (update device (nth->_t ) if (nth->compute_gpu ) async (nth->streams [nth->stream_id ]))
384
+ nrn_pragma_acc (update device (nth->_t ) if (nth->compute_gpu )
385
+ async (nth->streams [nth->stream_id ]))
381
386
nrn_pragma_acc (wait async (nth->streams [nth->stream_id ]))
382
387
nrn_pragma_omp (target update to (nth->_t ) if (nth->compute_gpu ))
383
388
fixed_play_continuous (nth);
0 commit comments