Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

some changes to avoid stack overflow on windows for large arrays #12

Merged
merged 3 commits into from
Feb 26, 2024
Merged

some changes to avoid stack overflow on windows for large arrays #12

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
9f32913
some changes to avoid stack overflow on windows for large arrays
jacobwilliams Feb 23, 2024
ed7529b
Merge branch 'master' of https://github.com/jacobwilliams/NumDiff int…
jacobwilliams Feb 23, 2024
b474a13
add a method to change the dpert array after initialization
jacobwilliams Feb 26, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
51 changes: 48 additions & 3 deletions src/numerical_differentiation_module.f90
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -222,6 +222,7 @@ module numerical_differentiation_module

procedure,public :: failed !! to check if an exception was raised.
procedure,public :: get_error_status !! the status of error condition
procedure,public :: set_dpert !! change the `dpert` values

! internal routines:
procedure :: destroy_sparsity_pattern !! destroy the sparsity pattern
Expand Down Expand Up @@ -1260,6 +1261,26 @@ subroutine set_numdiff_sparsity_bounds(me,xlow,xhigh)
end subroutine set_numdiff_sparsity_bounds
!*******************************************************************************

!*******************************************************************************
!>
! Change the `dpert` vector. Can be used after the class has been initialized
! to change the perturbation step sizes (e.g., after an iteration).

subroutine set_dpert(me,dpert)

class(numdiff_type),intent(inout) :: me
real(wp),dimension(:),intent(in) :: dpert !! perturbation vector for `x`

if (size(dpert)/=me%n) then
call me%raise_exception(29,'set_dpert',&
'incorrect size of dpert array')
else
me%dpert = abs(dpert) ! update
end if

end subroutine set_dpert
!*******************************************************************************

!*******************************************************************************
!>
! Initialize a [[numdiff_type]] class. This must be called first.
Expand Down Expand Up @@ -1606,7 +1627,17 @@ subroutine columns_in_partition_group(me,igroup,n_cols,cols,nonzero_rows,indices
num_nonzero_elements_in_col
if (allocated(col_indices)) deallocate(col_indices)
allocate(col_indices(num_nonzero_elements_in_col))
col_indices = pack(me%indices,mask=me%icol==cols(i))
!col_indices = pack(me%indices,mask=me%icol==cols(i))
block
integer :: j,n
n = 0
do j = 1, size(me%icol)
if (me%icol(j)==cols(i)) then
n = n + 1
col_indices(n) = j
end if
end do
end block
if (allocated(nonzero_rows)) then
nonzero_rows = [nonzero_rows,me%irow(col_indices)]
indices = [indices,col_indices]
Expand Down Expand Up @@ -1652,7 +1683,10 @@ subroutine compute_indices(me)
integer :: i !! counter

allocate(me%indices(me%num_nonzero_elements))
me%indices = [(i,i=1,me%num_nonzero_elements)]
!me%indices = [(i,i=1,me%num_nonzero_elements)]
do i = 1, me%num_nonzero_elements
me%indices(i) = i
end do

end subroutine compute_indices
!*******************************************************************************
Expand Down Expand Up @@ -2780,7 +2814,18 @@ subroutine compute_jacobian_partitioned(me,x,dx,jac)
num_nonzero_elements_in_col = count(me%sparsity%icol==cols(i))
if (allocated(col_indices)) deallocate(col_indices)
allocate(col_indices(num_nonzero_elements_in_col))
col_indices = pack(me%sparsity%indices,mask=me%sparsity%icol==cols(i))
! col_indices = pack(me%sparsity%indices,mask=me%sparsity%icol==cols(i))
block
integer :: j,n
n = 0
do j = 1, size(me%sparsity%icol)
if (me%sparsity%icol(j)==cols(i)) then
n = n + 1
col_indices(n) = j
end if
end do
end block

df(me%sparsity%irow(col_indices)) = df(me%sparsity%irow(col_indices)) / &
(me%meth(1)%df_den_factor*dx(cols(i)))
end do
Expand Down
Loading