LCPQ/quantum_package
10
0
Fork 0
mirror of https://github.com/LCPQ/quantum_package synced 2024-12-22 12:23:48 +01:00
Code Issues Releases Wiki Activity

mpif.h

Browse Source
This commit is contained in:
Anthony Scemama 2017-11-27 09:58:48 +01:00
parent 4243595a17
commit fc5f0b9745
1 changed files with 20 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
plugins/MPI/mpi.irp.f
View File

@ -1,11 +1,11 @@
BEGIN_PROVIDER [ integer, mpi_bit_kind ]
use bitmasks
include 'mpif.h'
implicit none
BEGIN_DOC
! MPI bit kind type
END_DOC
IRP_IF MPI
include 'mpif.h'
if (bit_kind == 4) then
mpi_bit_kind = MPI_INTEGER4
else if (bit_kind == 8) then
@ -89,10 +89,11 @@ subroutine broadcast_chunks_$double(A, LDA)
integer, intent(in) :: LDA
$type, intent(inout) :: A(LDA)
use bitmasks
include 'mpif.h'
BEGIN_DOC
! Broadcast with chunks of ~2GB
END_DOC
IRP_IF MPI
include 'mpif.h'
integer :: i, sze, ierr
do i=1,LDA,200000000/$8
sze = min(LDA-i+1, 200000000/$8)
@ -102,7 +103,7 @@ subroutine broadcast_chunks_$double(A, LDA)
stop -1
endif
enddo
IRP_ENDIF
end
SUBST [ double, type, 8, DOUBLE_PRECISION ]