@@ -438,20 +438,17 @@ __launch_bounds__(128) __global__
438438 DW_DTYPE const *__restrict__ const bz_grad_store_1,
439439 DW_DTYPE const *__restrict__ const azh,
440440 DW_DTYPE const *__restrict__ const bz,
441- DW_DTYPE const *__restrict__ const bzh,
442441#endif
443442#if DW_NDIM >= 2
444443 DW_DTYPE *__restrict__ const grad_by_shot,
445444 DW_DTYPE const *__restrict__ const by_grad_store_1,
446445 DW_DTYPE const *__restrict__ const ayh,
447446 DW_DTYPE const *__restrict__ const by,
448- DW_DTYPE const *__restrict__ const byh,
449447#endif
450448 DW_DTYPE *__restrict__ const grad_bx_shot,
451449 DW_DTYPE const *__restrict__ const bx_grad_store_1,
452450 DW_DTYPE const *__restrict__ const axh,
453451 DW_DTYPE const *__restrict__ const bx,
454- DW_DTYPE const *__restrict__ const bxh,
455452 bool const b_requires_grad) {
456453
457454#if DW_NDIM == 3
@@ -577,16 +574,13 @@ __launch_bounds__(128) __global__
577574#if DW_NDIM >= 3
578575 DW_DTYPE const *__restrict__ const az,
579576 DW_DTYPE const *__restrict__ const bzh,
580- DW_DTYPE const *__restrict__ const bz,
581577#endif
582578#if DW_NDIM >= 2
583579 DW_DTYPE const *__restrict__ const ay,
584580 DW_DTYPE const *__restrict__ const byh,
585- DW_DTYPE const *__restrict__ const by,
586581#endif
587582 DW_DTYPE const *__restrict__ const ax,
588583 DW_DTYPE const *__restrict__ const bxh,
589- DW_DTYPE const *__restrict__ const bx,
590584 bool const k_requires_grad) {
591585
592586#if DW_NDIM == 3
@@ -1550,12 +1544,12 @@ extern "C"
15501544#endif
15511545 buoyancy_x,
15521546#if DW_NDIM >= 3
1553- grad_bz_shot, bz_grad_store_1_t , azh, bz, bzh,
1547+ grad_bz_shot, bz_grad_store_1_t , azh, bz,
15541548#endif
15551549#if DW_NDIM >= 2
1556- grad_by_shot, by_grad_store_1_t , ayh, by, byh,
1550+ grad_by_shot, by_grad_store_1_t , ayh, by,
15571551#endif
1558- grad_bx_shot, bx_grad_store_1_t , axh, bx, bxh,
1552+ grad_bx_shot, bx_grad_store_1_t , axh, bx,
15591553 b_requires_grad && ((t % step_ratio_h) == 0 ));
15601554 CHECK_KERNEL_ERROR
15611555
@@ -1628,12 +1622,12 @@ extern "C"
16281622#endif
16291623 buoyancy_x, grad_k_shot, k_grad_store_1_t ,
16301624#if DW_NDIM >= 3
1631- az, bzh, bz,
1625+ az, bzh,
16321626#endif
16331627#if DW_NDIM >= 2
1634- ay, byh, by,
1628+ ay, byh,
16351629#endif
1636- ax, bxh, bx, k_grad_cond);
1630+ ax, bxh, k_grad_cond);
16371631 CHECK_KERNEL_ERROR
16381632
16391633 if (use_double_buffering) {
0 commit comments