Skip to content

Commit

Permalink
post-merge fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Jacques Xing committed Sep 11, 2024
1 parent 11113de commit e1e5ccd
Showing 1 changed file with 12 additions and 9 deletions.
21 changes: 12 additions & 9 deletions src/solver.f90
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ module m_solver
integer :: steps_between_checkpoints ! checkpoints can be used as restarts
integer :: steps_between_snapshots

class(field_t), pointer :: u, v, w, pressure
class(field_t), pointer :: u, v, w

class(base_backend_t), pointer :: backend
class(mesh_t), pointer :: mesh
Expand Down Expand Up @@ -127,7 +127,6 @@ function init(backend, mesh, host_allocator) result(solver)
solver%u => solver%backend%allocator%get_block(DIR_X, VERT)
solver%v => solver%backend%allocator%get_block(DIR_X, VERT)
solver%w => solver%backend%allocator%get_block(DIR_X, VERT)
solver%pressure => solver%backend%allocator%get_block(DIR_Z, CELL)

! set defaults
poisson_solver_type = 'FFT'
Expand All @@ -153,10 +152,10 @@ function init(backend, mesh, host_allocator) result(solver)

solver%steps_between_checkpoints = 1
solver%steps_between_snapshots = 1
solver%dt = globs%dt
solver%backend%nu = globs%nu
solver%n_iters = globs%n_iters
solver%n_output = globs%n_output
solver%dt = dt
solver%backend%nu = 1._dp/Re
solver%n_iters = n_iters
solver%n_output = n_output
solver%ngrid = product(solver%mesh%get_global_dims(VERT))

dims = solver%mesh%get_dims(VERT)
Expand Down Expand Up @@ -489,7 +488,7 @@ subroutine run(self)

class(solver_t), intent(inout) :: self

class(field_t), pointer :: du, dv, dw, div_u, dpdx, dpdy, dpdz
class(field_t), pointer :: du, dv, dw, div_u, pressure, dpdx, dpdy, dpdz
class(field_t), pointer :: u_out, v_out, w_out

integer(8), dimension(3) :: icount !! Local size of output array
Expand Down Expand Up @@ -529,15 +528,19 @@ subroutine run(self)

call self%divergence_v2p(div_u, self%u, self%v, self%w)

call self%poisson(self%pressure, div_u)
pressure => self%backend%allocator%get_block(DIR_Z, CELL)

call self%poisson(pressure, div_u)

call self%backend%allocator%release_block(div_u)

dpdx => self%backend%allocator%get_block(DIR_X)
dpdy => self%backend%allocator%get_block(DIR_X)
dpdz => self%backend%allocator%get_block(DIR_X)

call self%gradient_p2v(dpdx, dpdy, dpdz, self%pressure)
call self%gradient_p2v(dpdx, dpdy, dpdz, pressure)

call self%backend%allocator%release_block(pressure)

! velocity correction
call self%backend%vecadd(-1._dp, dpdx, 1._dp, self%u)
Expand Down

0 comments on commit e1e5ccd

Please sign in to comment.