| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969 |
- !> Provides a generic and simpler interface into MPI calls for OASIS.
- MODULE mod_oasis_mpi
- !-------------------------------------------------------------------------------
- ! PURPOSE: general layer on MPI functions
- !-------------------------------------------------------------------------------
- use mod_oasis_kinds
- USE mod_oasis_data ,ONLY: nulprt, OASIS_Debug
- USE mod_oasis_sys ,ONLY: oasis_debug_enter, oasis_debug_exit, oasis_flush, oasis_abort, astr
- USE mod_oasis_timer,ONLY: oasis_timer_start, oasis_timer_stop
- implicit none
- private
- ! PUBLIC: Public interfaces
- public :: oasis_mpi_chkerr
- public :: oasis_mpi_send
- public :: oasis_mpi_recv
- public :: oasis_mpi_bcast
- public :: oasis_mpi_gathScatVInit
- public :: oasis_mpi_gatherV
- public :: oasis_mpi_scatterV
- public :: oasis_mpi_sum
- public :: oasis_mpi_min
- public :: oasis_mpi_max
- public :: oasis_mpi_commsize
- public :: oasis_mpi_commrank
- public :: oasis_mpi_initialized
- public :: oasis_mpi_wtime
- public :: oasis_mpi_abort
- public :: oasis_mpi_barrier
- public :: oasis_mpi_init
- public :: oasis_mpi_finalize
- public :: oasis_mpi_reducelists
- !> Generic overloaded interface into MPI send
- interface oasis_mpi_send ; module procedure &
- oasis_mpi_sendi0, &
- oasis_mpi_sendi1, &
- oasis_mpi_sendr0, &
- oasis_mpi_sendr1, &
- oasis_mpi_sendr3
- end interface
- !> Generic overloaded interface into MPI receive
- interface oasis_mpi_recv ; module procedure &
- oasis_mpi_recvi0, &
- oasis_mpi_recvi1, &
- oasis_mpi_recvr0, &
- oasis_mpi_recvr1, &
- oasis_mpi_recvr3
- end interface
- !> Generic overloaded interface into MPI broadcast
- interface oasis_mpi_bcast ; module procedure &
- oasis_mpi_bcastc0, &
- oasis_mpi_bcastc1, &
- oasis_mpi_bcastl0, &
- oasis_mpi_bcastl1, &
- oasis_mpi_bcasti0, &
- oasis_mpi_bcasti1, &
- oasis_mpi_bcasti2, &
- oasis_mpi_bcastr0, &
- oasis_mpi_bcastr1, &
- oasis_mpi_bcastr2, &
- oasis_mpi_bcastr3
- end interface
- !> Generic interface to oasis_mpi_gathScatVInit
- interface oasis_mpi_gathScatVInit ; module procedure &
- oasis_mpi_gathScatVInitr1
- end interface
- !> Generic interfaces into an MPI vector gather
- interface oasis_mpi_gatherv ; module procedure &
- oasis_mpi_gatherVr1
- end interface
- !> Generic interfaces into an MPI vector scatter
- interface oasis_mpi_scatterv ; module procedure &
- oasis_mpi_scatterVr1
- end interface
- !> Generic overloaded interface into MPI sum reduction
- interface oasis_mpi_sum ; module procedure &
- oasis_mpi_sumi0, &
- oasis_mpi_sumi1, &
- oasis_mpi_sumb0, &
- oasis_mpi_sumb1, &
- oasis_mpi_sumr0, &
- oasis_mpi_sumr1, &
- oasis_mpi_sumr2, &
- oasis_mpi_sumr3
- end interface
- !> Generic overloaded interface into MPI min reduction
- interface oasis_mpi_min ; module procedure &
- oasis_mpi_mini0, &
- oasis_mpi_mini1, &
- oasis_mpi_minr0, &
- oasis_mpi_minr1
- end interface
- !> Generic overloaded interface into MPI max reduction
- interface oasis_mpi_max ; module procedure &
- oasis_mpi_maxi0, &
- oasis_mpi_maxi1, &
- oasis_mpi_maxr0, &
- oasis_mpi_maxr1
- end interface
- ! mpi library include file
- #include <mpif.h>
- !===============================================================================
- CONTAINS
- !===============================================================================
- !> Checks MPI error codes and aborts
- !> This method compares rcode to MPI_SUCCESS. If rcode is an error,
- !> it queries MPI_ERROR_STRING for the error string associated with rcode, writes
- !> it out, and aborts with the string passed through the interface.
- SUBROUTINE oasis_mpi_chkerr(rcode,string)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: rcode !< MPI error code
- character(*), intent(in) :: string !< abort message
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_chkerr)'
- character(MPI_MAX_ERROR_STRING) :: lstring
- integer(ip_i4_p) :: len
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: layer on MPI error checking
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lstring = ' '
- if (rcode /= MPI_SUCCESS) then
- call MPI_ERROR_STRING(rcode,lstring,len,ierr)
- call oasis_mpi_abort(subname//trim(string)//':'//trim(lstring),rcode)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_chkerr
- !===============================================================================
- !===============================================================================
- !> Send a scalar integer
- SUBROUTINE oasis_mpi_sendi0(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec !< send value
- integer(ip_i4_p), intent(in) :: pid !< pid to send to
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sendi0)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Send a single integer
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = 1
- call MPI_SEND(lvec,lsize,MPI_INTEGER,pid,tag,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sendi0
- !===============================================================================
- !===============================================================================
- !> Send an array of 1D integers
- SUBROUTINE oasis_mpi_sendi1(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec(:) !< send values
- integer(ip_i4_p), intent(in) :: pid !< pid to send to
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sendi1)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Send a vector of integers
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(lvec)
- call MPI_SEND(lvec,lsize,MPI_INTEGER,pid,tag,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sendi1
- !===============================================================================
- !===============================================================================
- !> Send a scalar double
- SUBROUTINE oasis_mpi_sendr0(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p),intent(in) :: lvec !< send values
- integer(ip_i4_p), intent(in) :: pid !< pid to send to
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sendr0)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Send a real scalar
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = 1
- call MPI_SEND(lvec,lsize,MPI_REAL8,pid,tag,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sendr0
- !===============================================================================
- !===============================================================================
- !> Send an array of 1D doubles
- SUBROUTINE oasis_mpi_sendr1(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p),intent(in) :: lvec(:) !< send values
- integer(ip_i4_p), intent(in) :: pid !< pid to send to
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sendr1)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Send a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(lvec)
- call MPI_SEND(lvec,lsize,MPI_REAL8,pid,tag,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sendr1
- !===============================================================================
- !===============================================================================
- !> Send an array of 3D doubles
- SUBROUTINE oasis_mpi_sendr3(array,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p),intent(in) :: array(:,:,:) !< send values
- integer(ip_i4_p), intent(in) :: pid !< pid to send to
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sendr3)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Send a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(array)
- call MPI_SEND(array,lsize,MPI_REAL8,pid,tag,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sendr3
- !===============================================================================
- !===============================================================================
- !> Receive a scalar integer
- SUBROUTINE oasis_mpi_recvi0(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(out):: lvec !< receive values
- integer(ip_i4_p), intent(in) :: pid !< pid to recv from
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_recvi0)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: status(MPI_STATUS_SIZE) ! mpi status info
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Recv a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = 1
- call MPI_RECV(lvec,lsize,MPI_INTEGER,pid,tag,comm,status,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_recvi0
- !===============================================================================
- !===============================================================================
- !> Receive an array of 1D integers
- SUBROUTINE oasis_mpi_recvi1(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(out):: lvec(:) !< receive values
- integer(ip_i4_p), intent(in) :: pid !< pid to recv from
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_recvi1)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: status(MPI_STATUS_SIZE) ! mpi status info
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Recv a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(lvec)
- call MPI_RECV(lvec,lsize,MPI_INTEGER,pid,tag,comm,status,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_recvi1
- !===============================================================================
- !===============================================================================
- !> Receive a scalar double
- SUBROUTINE oasis_mpi_recvr0(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p),intent(out):: lvec !< receive values
- integer(ip_i4_p), intent(in) :: pid !< pid to recv from
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_recvr0)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: status(MPI_STATUS_SIZE) ! mpi status info
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Recv a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = 1
- call MPI_RECV(lvec,lsize,MPI_REAL8,pid,tag,comm,status,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_recvr0
- !===============================================================================
- !===============================================================================
- !> Receive an array of 1D doubles
- SUBROUTINE oasis_mpi_recvr1(lvec,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p),intent(out):: lvec(:) !< receive values
- integer(ip_i4_p), intent(in) :: pid !< pid to recv from
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_recvr1)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: status(MPI_STATUS_SIZE) ! mpi status info
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Recv a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(lvec)
- call MPI_RECV(lvec,lsize,MPI_REAL8,pid,tag,comm,status,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_recvr1
- !===============================================================================
- !===============================================================================
- !> Receive an array of 3D doubles
- SUBROUTINE oasis_mpi_recvr3(array,pid,tag,comm,string)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p),intent(out):: array(:,:,:) !< receive values
- integer(ip_i4_p), intent(in) :: pid !< pid to recv from
- integer(ip_i4_p), intent(in) :: tag !< tag
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_recvr3)'
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: status(MPI_STATUS_SIZE) ! mpi status info
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Recv a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(array)
- call MPI_RECV(array,lsize,MPI_REAL8,pid,tag,comm,status,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_recvr3
- !===============================================================================
- !===============================================================================
- !> Broadcast a scalar integer
- SUBROUTINE oasis_mpi_bcasti0(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(inout):: vec !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcasti0)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast an integer
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = 1
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_INTEGER,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcasti0
- !===============================================================================
- !===============================================================================
- !> Broadcast a scalar logical
- SUBROUTINE oasis_mpi_bcastl0(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- logical, intent(inout):: vec !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcastl0)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a logical
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = 1
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_LOGICAL,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastl0
- !===============================================================================
- !===============================================================================
- !> Broadcast a character string
- SUBROUTINE oasis_mpi_bcastc0(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- character(len=*), intent(inout):: vec !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcastc0)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a character string
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = len(vec)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_CHARACTER,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastc0
- !===============================================================================
- !===============================================================================
- !> Broadcast an array of 1D character strings
- SUBROUTINE oasis_mpi_bcastc1(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- character(len=*), intent(inout):: vec(:) !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcastc1)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a character string
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(vec)*len(vec)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_CHARACTER,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastc1
- !===============================================================================
- !===============================================================================
- !> Broadcast a scalar double
- SUBROUTINE oasis_mpi_bcastr0(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(inout):: vec !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcastr0)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a real
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = 1
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_REAL8,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastr0
- !===============================================================================
- !===============================================================================
- !> Broadcast an array of 1D integers
- SUBROUTINE oasis_mpi_bcasti1(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(inout):: vec(:) !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcasti1)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a vector of integers
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(vec)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_INTEGER,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcasti1
- !===============================================================================
- !===============================================================================
- !> Broadcast an array of 1D logicals
- SUBROUTINE oasis_mpi_bcastl1(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- logical, intent(inout):: vec(:) !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcastl1)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a logical
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(vec)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_LOGICAL,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastl1
- !===============================================================================
- !===============================================================================
- !> Broadcast an array of 1D doubles
- SUBROUTINE oasis_mpi_bcastr1(vec,comm,string,pebcast)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(inout):: vec(:) !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_bcastr1)'
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a vector of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(vec)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(vec,lsize,MPI_REAL8,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastr1
- !===============================================================================
- !===============================================================================
- !> Broadcast an array of 2D doubles
- SUBROUTINE oasis_mpi_bcastr2(arr,comm,string,pebcast)
- IMPLICIT none
- !----- arguments -----
- real(ip_double_p), intent(inout):: arr(:,:) !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local -----
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !----- formats -----
- character(*),parameter :: subname = '(oasis_mpi_bcastr2)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a 2d array of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(arr)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(arr,lsize,MPI_REAL8,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastr2
- !===============================================================================
- !===============================================================================
- !> Broadcast an array of 2D integers
- SUBROUTINE oasis_mpi_bcasti2(arr,comm,string,pebcast)
- IMPLICIT none
- !----- arguments -----
- integer, intent(inout):: arr(:,:) !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local -----
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !----- formats -----
- character(*),parameter :: subname = '(oasis_mpi_bcasti2)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a 2d array of integers
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(arr)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(arr,lsize,MPI_INTEGER,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcasti2
- !===============================================================================
- !===============================================================================
- !> Broadcast an array of 3D doubles
- SUBROUTINE oasis_mpi_bcastr3(arr,comm,string,pebcast)
- IMPLICIT none
- !----- arguments -----
- real(ip_double_p), intent(inout):: arr(:,:,:) !< values to broadcast
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- integer(ip_i4_p), optional, intent(in) :: pebcast !< bcast pe, default is task 0
- !----- local -----
- integer(ip_i4_p) :: ierr
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: lpebcast
- !----- formats -----
- character(*),parameter :: subname = '(oasis_mpi_bcastr3)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: Broadcast a 3d array of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- lsize = size(arr)
- lpebcast = 0
- if (present(pebcast)) lpebcast = pebcast
- call MPI_BCAST(arr,lsize,MPI_REAL8,lpebcast,comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_bcastr3
- !===============================================================================
- !===============================================================================
- !> Initialize variables for oasis_mpi_gatherv and oasis_mpi_scatterv
- !> This method initializes glob1DArr, globSize, and displs for use
- !> in the oasis_mpi_gatherv and oasis_mpi_scatterv routines. locArr is the
- !> distributed array to gather from or scatter to.
- SUBROUTINE oasis_mpi_gathScatvInitr1(comm, rootid, locArr, glob1DArr, globSize, &
- displs, string )
- IMPLICIT none
- !----- arguments -----
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- integer(ip_i4_p), intent(in) :: rootid !< MPI task to gather/scatter on
- real(ip_double_p),intent(in) :: locArr(:) !< Local array of distributed data
- real(ip_double_p),pointer :: glob1DArr(:) !< Global 1D array of gathered data
- integer(ip_i4_p), pointer :: globSize(:) !< Size of each distributed piece
- integer(ip_i4_p), pointer :: displs(:) !< Displacements for receive
- character(*),optional,intent(in):: string !< to identify caller
- !----- local -----
- integer(ip_i4_p) :: npes ! Number of MPI tasks
- integer(ip_i4_p) :: locSize ! Size of local distributed data
- integer(ip_i4_p), pointer :: sendSize(:) ! Size to send for initial gather
- integer(ip_i4_p) :: i ! Index
- integer(ip_i4_p) :: rank ! Rank of this MPI task
- integer(ip_i4_p) :: nSize ! Maximum size to send
- integer(ip_i4_p) :: ierr ! Error code
- integer(ip_i4_p) :: nSiz1D ! Size of 1D global array
- integer(ip_i4_p) :: maxSize ! Maximum size
- !----- formats -----
- character(*),parameter :: subname = '(oasis_mpi_gathScatvInitr1)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: Setup arrays for a gatherv/scatterv operation
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- locSize = size(locarr)
- call oasis_mpi_commsize( comm, npes )
- call oasis_mpi_commrank( comm, rank )
- allocate( globSize(npes) )
- !
- ! --- Gather the send global sizes from each MPI task -----------------------
- !
- allocate( sendSize(npes) )
- sendSize(:) = 1
- globSize(:) = 1
- call MPI_GATHER( locSize, 1, MPI_INTEGER, globSize, sendSize, &
- MPI_INTEGER, rootid, comm, ierr )
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- deallocate( sendSize )
- !
- ! --- Prepare the displacement and allocate arrays -------------------------
- !
- allocate( displs(npes) )
- displs(1) = 0
- if ( rootid /= rank )then
- maxSize = 1
- globSize = 1
- else
- maxSize = maxval(globSize)
- end if
- nsiz1D = min(maxSize,globSize(1))
- do i = 2, npes
- nSize = min(maxSize,globSize(i-1))
- displs(i) = displs(i-1) + nSize
- nsiz1D = nsiz1D + min(maxSize,globSize(i))
- end do
- allocate( glob1DArr(nsiz1D) )
- !----- Do some error checking for the root task arrays computed ----
- if ( rootid == rank )then
- if ( nsiz1D /= sum(globSize) ) &
- call oasis_mpi_abort( subName//" : Error, size of global array not right" )
- if ( any(displs < 0) .or. any(displs >= nsiz1D) ) &
- call oasis_mpi_abort( subName//" : Error, displacement array not right" )
- if ( (displs(npes)+globSize(npes)) /= nsiz1D ) &
- call oasis_mpi_abort( subName//" : Error, displacement array values too big" )
- end if
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_gathScatvInitr1
- !===============================================================================
- !===============================================================================
- !> Gather a vector of distributed data to a rootid
- !> This method passes in glob1DArr, globSize, and displs computed
- !> in the oasis_mpi_gathscatvinit routine and uses that information to
- !> gather the locArr into the glob1Darr on processor rootid in communicator
- !> comm.
- SUBROUTINE oasis_mpi_gathervr1(locarr, locSize, glob1DArr, globSize, displs, rootid, &
- comm, string )
- IMPLICIT none
- !----- arguments -----
- real(ip_double_p),intent(in) :: locArr(:) !< Local array
- real(ip_double_p),intent(inout) :: glob1DArr(:) !< Global 1D array to receive in on
- integer(ip_i4_p), intent(in) :: locSize !< Number to send from this PE
- integer(ip_i4_p), intent(in) :: globSize(:) !< Number to receive from each PE
- integer(ip_i4_p), intent(in) :: displs(:) !< Displacements for receives
- integer(ip_i4_p), intent(in) :: rootid !< MPI task to gather on
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in):: string !< to identify caller
- !----- local -----
- integer(ip_i4_p) :: ierr ! Error code
- !----- formats -----
- character(*),parameter :: subname = '(oasis_mpi_gathervr1)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: Gather a 1D array of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_GATHERV( locarr, locSize, MPI_REAL8, glob1Darr, globSize, displs, &
- MPI_REAL8, rootid, comm, ierr )
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_gathervr1
- !===============================================================================
- !===============================================================================
- !> Scatter a vector of global data from a rootid
- !> This method passes in glob1DArr, globSize, and displs computed
- !> in the oasis_mpi_gathscatvinit routine and uses that information to
- !> scatter glob1Darr on processor rootid in communicator comm to locarr
- !> on other processors.
- SUBROUTINE oasis_mpi_scattervr1(locarr, locSize, glob1Darr, globSize, displs, rootid, &
- comm, string )
- IMPLICIT none
- !----- arguments -----
- real(ip_double_p),intent(out) :: locarr(:) !< Local array
- real(ip_double_p),intent(in) :: glob1Darr(:) !< Global 1D array to send from
- integer(ip_i4_p), intent(in) :: locSize !< Number to receive this PE
- integer(ip_i4_p), intent(in) :: globSize(:) !< Number to send to each PE
- integer(ip_i4_p), intent(in) :: displs(:) !< Displacements for send
- integer(ip_i4_p), intent(in) :: rootid !< MPI task to scatter on
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in):: string !< to identify caller
- !----- local -----
- integer(ip_i4_p) :: ierr ! Error code
- !----- formats -----
- character(*),parameter :: subname = '(oasis_mpi_scattervr1)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: Scatter a 1D array of reals
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_SCATTERV( glob1Darr, globSize, displs, MPI_REAL8, locarr, locSize, &
- MPI_REAL8, rootid, comm, ierr )
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_scattervr1
- !===============================================================================
- !===============================================================================
- !> Compute a global Sum for a scalar integer
- SUBROUTINE oasis_mpi_sumi0(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec !< local values
- integer(ip_i4_p), intent(out):: gvec !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumi0)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = 1
- gsize = 1
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumi0
- !===============================================================================
- !===============================================================================
- !> Compute a 1D array of global sums for an array of 1D integers
- !> This sums an array of local integers to an array of summed integers.
- !> This does not reduce the array to a scalar.
- SUBROUTINE oasis_mpi_sumi1(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec(:) !< local values
- integer(ip_i4_p), intent(out):: gvec(:) !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumi1)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumi1
- !===============================================================================
- !===============================================================================
- !> Compute a global sum for a scalar 8 byte integer
- SUBROUTINE oasis_mpi_sumb0(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i8_p), intent(in) :: lvec !< local values
- integer(ip_i8_p), intent(out):: gvec !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumb0)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = 1
- gsize = 1
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumb0
- !===============================================================================
- !===============================================================================
- !> Compute a 1D array of global sums for an array of 1D 8 byte integers
- !> This sums an array of local integers to an array of summed integers.
- !> This does not reduce the array to a scalar.
- SUBROUTINE oasis_mpi_sumb1(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i8_p), intent(in) :: lvec(:) !< local values
- integer(ip_i8_p), intent(out):: gvec(:) !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumb1)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumb1
- !===============================================================================
- !===============================================================================
- !> Compute a global sum for a scalar double
- SUBROUTINE oasis_mpi_sumr0(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec !< local values
- real(ip_double_p), intent(out):: gvec !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumr0)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = 1
- gsize = 1
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumr0
- !===============================================================================
- !===============================================================================
- !> Compute a 1D array of global sums for an array of 1D doubles
- !> This sums an array of local doubles to an array of summed doubles.
- !> This does not reduce the array to a scalar.
- SUBROUTINE oasis_mpi_sumr1(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec(:) !< local values
- real(ip_double_p), intent(out):: gvec(:) !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumr1)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumr1
- !===============================================================================
- !===============================================================================
- !> Compute a 2D array of global sums for an array of 2D doubles
- !> This sums an array of local doubles to an array of summed doubles.
- !> This does not reduce the array to a scalar.
- SUBROUTINE oasis_mpi_sumr2(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec(:,:)!< local values
- real(ip_double_p), intent(out):: gvec(:,:)!< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumr2)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumr2
- !===============================================================================
- !===============================================================================
- !> Compute a 3D array of global sums for an array of 3D doubles
- !> This sums an array of local doubles to an array of summed doubles.
- !> This does not reduce the array to a scalar.
- SUBROUTINE oasis_mpi_sumr3(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec(:,:,:) !< local values
- real(ip_double_p), intent(out):: gvec(:,:,:) !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_sumr3)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds sum of a distributed vector of values, assume local sum
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_SUM
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_sumr3
- !===============================================================================
- !===============================================================================
- !> Compute a global minimum for a scalar integer
- SUBROUTINE oasis_mpi_mini0(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec !< local values
- integer(ip_i4_p), intent(out):: gvec !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_mini0)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds min of a distributed vector of values, assume local min
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MIN
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = 1
- gsize = 1
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_mini0
- !===============================================================================
- !===============================================================================
- !> Compute an array of global minimums for an array of 1D integers
- SUBROUTINE oasis_mpi_mini1(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec(:) !< local values
- integer(ip_i4_p), intent(out):: gvec(:) !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_mini1)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds min of a distributed vector of values, assume local min
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MIN
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_mini1
- !===============================================================================
- !===============================================================================
- !> Compute an global minimum for a scalar double
- SUBROUTINE oasis_mpi_minr0(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec !< local values
- real(ip_double_p), intent(out):: gvec !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_minr0)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds min of a distributed vector of values, assume local min
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MIN
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = 1
- gsize = 1
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_minr0
- !===============================================================================
- !===============================================================================
- !> Compute an array of global minimums for an array of 1D doubles
- SUBROUTINE oasis_mpi_minr1(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec(:) !< local values
- real(ip_double_p), intent(out):: gvec(:) !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_minr1)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds min of a distributed vector of values, assume local min
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MIN
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_minr1
- !===============================================================================
- !===============================================================================
- !> Compute a global maximum for a scalar integer
- SUBROUTINE oasis_mpi_maxi0(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec !< local values
- integer(ip_i4_p), intent(out):: gvec !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_maxi0)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds max of a distributed vector of values, assume local max
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MAX
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = 1
- gsize = 1
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_maxi0
- !===============================================================================
- !===============================================================================
- !> Compute an array of global maximums for an array of 1D integers
- SUBROUTINE oasis_mpi_maxi1(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- integer(ip_i4_p), intent(in) :: lvec(:) !< local values
- integer(ip_i4_p), intent(out):: gvec(:) !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_maxi1)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds max of a distributed vector of values, assume local max
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MAX
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_INTEGER,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_maxi1
- !===============================================================================
- !===============================================================================
- !> Compute a global maximum for a scalar double
- SUBROUTINE oasis_mpi_maxr0(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec !< local values
- real(ip_double_p), intent(out):: gvec !< global values
- integer(ip_i4_p), intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_maxr0)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds max of a distributed vector of values, assume local max
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MAX
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = 1
- gsize = 1
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_maxr0
- !===============================================================================
- !===============================================================================
- !> Compute an array of global maximums for an array of 1D doubles
- SUBROUTINE oasis_mpi_maxr1(lvec,gvec,comm,string,all)
- IMPLICIT none
- !----- arguments ---
- real(ip_double_p), intent(in) :: lvec(:) !< local values
- real(ip_double_p), intent(out):: gvec(:) !< global values
- integer(ip_i4_p) , intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- logical, optional,intent(in) :: all !< if true call allreduce, otherwise reduce to task 0
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_maxr1)'
- logical :: lall
- character(len=256) :: lstring
- integer(ip_i4_p) :: reduce_type ! mpi reduction type
- integer(ip_i4_p) :: lsize
- integer(ip_i4_p) :: gsize
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: Finds max of a distributed vector of values, assume local max
- ! already computed
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- reduce_type = MPI_MAX
- if (present(all)) then
- lall = all
- else
- lall = .false.
- endif
- if (present(string)) then
- lstring = trim(subName)//":"//trim(string)
- else
- lstring = trim(subName)
- endif
- lsize = size(lvec)
- gsize = size(gvec)
- if (lsize /= gsize) then
- call oasis_mpi_abort(subName//" lsize,gsize incompatable "//trim(string))
- endif
- if (lall) then
- call MPI_ALLREDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_ALLREDUCE")
- else
- call MPI_REDUCE(lvec,gvec,gsize,MPI_REAL8,reduce_type,0,comm,ierr)
- call oasis_mpi_chkerr(ierr,trim(lstring)//" MPI_REDUCE")
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_maxr1
- !===============================================================================
- !===============================================================================
- !> Get the total number of tasks associated with a communicator
- SUBROUTINE oasis_mpi_commsize(comm,size,string)
- IMPLICIT none
- !----- arguments ---
- integer,intent(in) :: comm !< mpi communicator
- integer,intent(out) :: size !< output comm size
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_commsize)'
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI commsize
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_COMM_SIZE(comm,size,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_commsize
- !===============================================================================
- !===============================================================================
- !> Get the rank (task ID) for a task in a communicator
- SUBROUTINE oasis_mpi_commrank(comm,rank,string)
- IMPLICIT none
- !----- arguments ---
- integer,intent(in) :: comm !< mpi communicator
- integer,intent(out) :: rank !< output task ID
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_commrank)'
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI commrank
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_COMM_RANK(comm,rank,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_commrank
- !===============================================================================
- !===============================================================================
- !> Check whether MPI has been initialized
- SUBROUTINE oasis_mpi_initialized(flag,string)
- IMPLICIT none
- !----- arguments ---
- logical,intent(out) :: flag !< true if MPI_INITIALIZED has been called
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subName = '(oasis_mpi_initialized)'
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI initialized
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_INITIALIZED(flag,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_initialized
- !===============================================================================
- !===============================================================================
- !> Return a timestamp from MPI_WTIME
- SUBROUTINE oasis_mpi_wtime(wtime)
- IMPLICIT none
- !----- arguments ---
- real(ip_r8_p), intent(out) :: wtime !< time in MPI_WTIME units
- !----- local ---
- character(*),parameter :: subName = '(oasis_mpi_wtime)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI wtime
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- wtime = MPI_WTIME()
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_wtime
- !===============================================================================
- !===============================================================================
- !> Write error messages and Call MPI_ABORT
- SUBROUTINE oasis_mpi_abort(string,rcode)
- IMPLICIT none
- !----- arguments ---
- character(*),optional,intent(in) :: string !< to identify caller
- integer,optional,intent(in) :: rcode !< optional code
- !----- local ---
- character(*),parameter :: subName = '(oasis_mpi_abort)'
- character(len=256) :: lstr
- integer(ip_i4_p) :: ierr
- integer :: rc ! return code
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI abort
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- if ( present(string) .and. present(rcode)) then
- write(lstr,'(a,i6.6)') trim(string)//' rcode = ',rcode
- elseif (present(string)) then
- lstr = trim(string)
- else
- lstr = ' '
- endif
- IF ( PRESENT(rcode)) THEN
- CALL oasis_abort(cd_routine=subName,cd_message=TRIM(string),rcode=rcode)
- ELSE
- CALL oasis_abort(cd_routine=subName,cd_message=TRIM(string))
- ENDIF
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_abort
- !===============================================================================
- !===============================================================================
- !> Call MPI_BARRIER for a particular communicator
- SUBROUTINE oasis_mpi_barrier(comm,string)
- IMPLICIT none
- !----- arguments ---
- integer,intent(in) :: comm !< mpi communicator
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_barrier)'
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI barrier
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_BARRIER(comm,ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_barrier
- !===============================================================================
- !===============================================================================
- !> Call MPI_INIT
- SUBROUTINE oasis_mpi_init(string)
- IMPLICIT none
- !----- arguments ---
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_init)'
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI init
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_INIT(ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_init
- !===============================================================================
- !===============================================================================
- !> Call MPI_FINALZE
- SUBROUTINE oasis_mpi_finalize(string)
- IMPLICIT none
- !----- arguments ---
- character(*),optional,intent(in) :: string !< to identify caller
- !----- local ---
- character(*),parameter :: subname = '(oasis_mpi_finalize)'
- integer(ip_i4_p) :: ierr
- !-------------------------------------------------------------------------------
- ! PURPOSE: MPI finalize
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- call MPI_FINALIZE(ierr)
- if (present(string)) then
- call oasis_mpi_chkerr(ierr,subName//trim(string))
- else
- call oasis_mpi_chkerr(ierr,subName)
- endif
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_finalize
- !===============================================================================
- !===============================================================================
- !> Custom method for reducing MPI lists across pes for OASIS
- SUBROUTINE oasis_mpi_reducelists(linp1,comm,cntout,lout1,callstr,fastcheck,fastcheckout, &
- linp2,lout2,spval2,linp3,lout3,spval3)
- IMPLICIT none
- !----- arguments ---
- character(*),pointer,intent(in) :: linp1(:) !< input list on each task
- integer ,intent(in) :: comm !< mpi communicator
- integer ,intent(out) :: cntout !< size of lout1 list
- character(*),pointer,intent(inout) :: lout1(:) !< reduced output list, same on all tasks
- character(*) ,intent(in) :: callstr !< to identify caller
- logical ,intent(in) ,optional :: fastcheck !< run a fastcheck first
- logical ,intent(out) ,optional :: fastcheckout !< true if fastcheck worked
- character(*),pointer,intent(in) ,optional :: linp2(:) !< input list on each task
- character(*),pointer,intent(inout),optional :: lout2(:) !< reduced output list, same on all tasks
- character(*) ,intent(in) ,optional :: spval2 !< unset value for linp2
- integer ,pointer,intent(in) ,optional :: linp3(:) !< input list on each task
- integer ,pointer,intent(inout),optional :: lout3(:) !< reduced output list, same on all tasks
- integer ,intent(in) ,optional :: spval3 !< unset value for linp3
- !----- local ---
- integer(kind=ip_i4_p) :: m,n,k,p
- integer(kind=ip_i4_p) :: llen,lsize
- integer(kind=ip_i4_p) :: cnt, cntr
- integer(kind=ip_i4_p) :: commrank, commsize
- integer(kind=ip_i4_p) :: listcheck, listcheckall
- integer(kind=ip_i4_p) :: maxloops, sendid, recvid, kfac
- logical :: found, present2, present3
- integer(kind=ip_i4_p) :: status(MPI_STATUS_SIZE) ! mpi status info
- character(len=ic_lvar2),pointer :: recv_varf1(:),varf1a(:),varf1b(:)
- character(len=ic_lvar2),pointer :: recv_varf2(:),varf2a(:),varf2b(:)
- integer(kind=ip_i4_p) ,pointer :: recv_varf3(:),varf3a(:),varf3b(:)
- character(len=ic_lvar2) :: string
- logical, parameter :: local_timers_on = .false.
- integer(ip_i4_p) :: ierr
- character(*),parameter :: subname = '(oasis_mpi_reducelists)'
- !-------------------------------------------------------------------------------
- ! PURPOSE: Custom method for reducing MPI lists for OASIS using a log2
- ! algorithm. This generates a list on all tasks that consists of the intersection
- ! of all the values on all the tasks with each value listed once. linp1
- ! is the input list, possibly different on each task. lout1
- ! is the resulting list, the same on each task, consistenting of all unique
- ! values of linp1 from all tasks. This ultimately reduces the list onto
- ! the root task and then it's broadcast. The reduction occurs via a binary
- ! type reduction from tasks to other tasks.
- !-------------------------------------------------------------------------------
- call oasis_debug_enter(subname)
- string = trim(callstr)
- if (present(fastcheckout)) fastcheckout = .false. ! by default
- call oasis_mpi_commrank(comm,commrank,string=subname//trim(string))
- call oasis_mpi_commsize(comm,commsize,string=subname//trim(string))
- !-----------------------------------------------
- !> * Check argument consistency
- !-----------------------------------------------
- if ((present(linp2) .and. .not.present(lout2)) .or. &
- (present(lout2) .and. .not.present(linp2))) then
- call oasis_mpi_abort(subname//trim(string)//" linp2 lout2 both must be present ")
- endif
- present2 = present(linp2)
- if ((present(linp3) .and. .not.present(lout3)) .or. &
- (present(lout3) .and. .not.present(linp3))) then
- call oasis_mpi_abort(subname//trim(string)//" linp3 lout3 both must be present ")
- endif
- present3 = present(linp3)
- if (len(linp1) > len(varf1a)) then
- call oasis_mpi_abort(subname//trim(string)//" linp1 too long ")
- endif
- if (present(linp2)) then
- if (size(linp2) /= size(linp1)) then
- call oasis_mpi_abort(subname//trim(string)//" linp1 linp2 not same size ")
- endif
- if (len(linp2) > len(varf2a)) then
- call oasis_mpi_abort(subname//trim(string)//" linp2 too long ")
- endif
- if (len(varf1a) /= len(varf2a)) then
- call oasis_mpi_abort(subname//trim(string)//" varf1a varf2a not same len ")
- endif
- endif
- if (present(linp3)) then
- if (size(linp3) /= size(linp1)) then
- call oasis_mpi_abort(subname//trim(string)//" linp1 linp3 not same size ")
- endif
- endif
- !-----------------------------------------------
- !> * Fast compare on all tasks
- ! If all tasks have same list, just skip the reduction
- !-----------------------------------------------
- if (present(fastcheck)) then
- if (fastcheck) then
- if (local_timers_on) call oasis_timer_start(trim(string)//'_rl_fastcheck')
- lsize = -1
- if (commrank == 0) then
- lsize = size(linp1)
- endif
- call oasis_mpi_bcast(lsize, comm, subname//trim(string)//' lsize check')
- ! varf1a holds linp1 from root on all tasks
- allocate(varf1a(lsize))
- varf1a = ' '
- if (commrank == 0) then
- varf1a(1:lsize) = linp1(1:lsize)
- endif
- call oasis_mpi_bcast(varf1a, comm, subname//trim(string)//' varf1a check')
- listcheck = 1
- if (OASIS_DEBUG >= 20) then
- write(nulprt,*) subname//trim(string),' sizes ',lsize,size(linp1)
- endif
- if (lsize /= size(linp1)) listcheck = 0
- n = 0
- do while (listcheck == 1 .and. n < lsize)
- n = n + 1
- if (varf1a(n) /= linp1(n)) listcheck = 0
- if (OASIS_DEBUG >= 20) then
- write(nulprt,*) subname//trim(string),' fcheck varf1a ',n,trim(linp1(n)),' ',trim(linp1(n)),listcheck
- endif
- enddo
- deallocate(varf1a)
- call oasis_mpi_min(listcheck,listcheckall,comm, subname//trim(string)//' listcheck',all=.true.)
- if (OASIS_DEBUG >= 15) then
- write(nulprt,*) subname//trim(string),' listcheck = ',listcheck,listcheckall
- endif
- if (local_timers_on) call oasis_timer_stop(trim(string)//'_rl_fastcheck')
- !-------------------------------------------------
- ! linp1 same on all tasks, update lout1, lout2, lout3 and return
- !-------------------------------------------------
- if (listcheckall == 1) then
- cntout = lsize
- allocate(lout1(lsize))
- lout1(1:lsize) = linp1(1:lsize)
- if (present2) then
- allocate(lout2(lsize))
- lout2(1:lsize) = linp2(1:lsize)
- endif
- if (present3) then
- allocate(lout3(lsize))
- lout3(1:lsize) = linp3(1:lsize)
- endif
- call oasis_debug_exit(subname)
- if (present(fastcheckout)) fastcheckout = .true.
- return
- endif
- endif ! fastcheck
- endif ! present fastcheck
- !-----------------------------------------------
- !> * Generate initial unique local name list
- !-----------------------------------------------
- llen = len(linp1)
- lsize = size(linp1)
- if (OASIS_Debug >= 15) then
- write(nulprt,*) subname//trim(string),' len, size = ',llen,lsize
- call oasis_flush(nulprt)
- endif
- allocate(varf1a(max(lsize,20))) ! 20 is arbitrary starting number
- if (present2) allocate(varf2a(max(lsize,20))) ! 20 is arbitrary starting number
- if (present3) allocate(varf3a(max(lsize,20))) ! 20 is arbitrary starting number
- cnt = 0
- do n = 1,lsize
- p = 0
- found = .false.
- do while (p < cnt .and. .not.found)
- p = p + 1
- if (linp1(n) == varf1a(p)) found = .true.
- enddo
- if (.not.found) then
- cnt = cnt + 1
- varf1a(cnt) = linp1(n)
- if (present2) varf2a(cnt) = linp2(n)
- if (present3) varf3a(cnt) = linp3(n)
- endif
- enddo
- !-----------------------------------------------
- !> * Log2 reduction of linp over tasks to root
- !-----------------------------------------------
- maxloops = int(sqrt(float(commsize+1)))+1
- do m = 1,maxloops
- kfac = 2**m
- recvid = commrank + kfac/2 ! task to recv from
- if (mod(commrank,kfac) /= 0 .or. &
- recvid < 0 .or. recvid > commsize-1) &
- recvid = -1
- sendid = commrank - kfac/2 ! task to send to
- if (mod(commrank+kfac/2,kfac) /= 0 .or. &
- sendid < 0 .or. sendid > commsize-1) &
- sendid = -1
- if (OASIS_Debug >= 15) then
- write(nulprt,*) subname//trim(string),' send/recv ids ',m,commrank,sendid,recvid
- call oasis_flush(nulprt)
- endif
- !-----------------------------------------------
- !> * Send list
- !-----------------------------------------------
- if (sendid >= 0) then
- if (local_timers_on) call oasis_timer_start(trim(string)//'_rl_send')
- call MPI_SEND(cnt, 1, MPI_INTEGER, sendid, 5900+m, comm, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':send cnt')
- if (cnt > 0) then
- if (OASIS_Debug >= 15) then
- write(nulprt,*) subname//trim(string),' send size ',commrank,m,cnt,ic_lvar2
- call oasis_flush(nulprt)
- endif
- call MPI_SEND(varf1a(1:cnt), cnt*ic_lvar2, MPI_CHARACTER, sendid, 6900+m, comm, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':send varf1a')
- if (present2) then
- call MPI_SEND(varf2a(1:cnt), cnt*ic_lvar2, MPI_CHARACTER, sendid, 7900+m, comm, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':send varf2a')
- endif
- if (present3) then
- call MPI_SEND(varf3a(1:cnt), cnt, MPI_INTEGER, sendid, 8900+m, comm, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':send varf3a')
- endif
- endif ! cnt > 0
- if (local_timers_on) call oasis_timer_stop (trim(string)//'_rl_send')
- endif ! sendid >= 0
- !-----------------------------------------------
- !> * Recv list
- !> * Determine the unique list
- !-----------------------------------------------
- if (recvid >= 0) then
- if (local_timers_on) call oasis_timer_start (trim(string)//'_rl_recv')
- call MPI_RECV(cntr, 1, MPI_INTEGER, recvid, 5900+m, comm, status, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':recv cntr')
- if (cntr > 0) then
- if (OASIS_Debug >= 15) then
- write(nulprt,*) subname//trim(string),' recv size ',commrank,m,cntr,ic_lvar2
- call oasis_flush(nulprt)
- endif
- allocate(recv_varf1(cntr))
- call MPI_RECV(recv_varf1, cntr*ic_lvar2, MPI_CHARACTER, recvid, 6900+m, comm, status, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':recv varf1')
- if (present2) then
- allocate(recv_varf2(cntr))
- call MPI_RECV(recv_varf2, cntr*ic_lvar2, MPI_CHARACTER, recvid, 7900+m, comm, status, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':recv varf2')
- endif
- if (present3) then
- allocate(recv_varf3(cntr))
- call MPI_RECV(recv_varf3, cntr, MPI_INTEGER, recvid, 8900+m, comm, status, ierr)
- call oasis_mpi_chkerr(ierr,subname//trim(string)//':recv varf3')
- endif
- endif ! cntr > 0
- if (local_timers_on) call oasis_timer_stop (trim(string)//'_rl_recv')
- if (local_timers_on) call oasis_timer_start(trim(string)//'_rl_rootsrch')
- do n = 1,cntr
- if (OASIS_Debug >= 15) write(nulprt,*) subname//trim(string),' check recv_varf1 ',m,n,trim(recv_varf1(n))
- p = 0
- found = .false.
- do while (p < cnt .and. .not.found)
- p = p + 1
- if (recv_varf1(n) == varf1a(p)) then
- found = .true.
- if (present2) then
- if (present(spval2)) then
- !--- use something other than spval2 if it exists and check consistency
- if (varf2a(p) == spval2) then
- varf2a(p) = recv_varf2(n)
- elseif (recv_varf2(n) /= spval2 .and. varf2a(p) /= recv_varf2(n)) then
- call oasis_abort(cd_routine=subname//trim(string),cd_message= &
- 'inconsistent linp2 value: '//trim(recv_varf2(n))//':'//trim(varf1a(p))//':'//trim(varf2a(p)))
- endif
- else
- if (varf2a(p) /= recv_varf2(n)) then
- call oasis_abort(cd_routine=subname//trim(string),cd_message= &
- 'inconsistent linp2 value: '//trim(recv_varf2(n))//':'//trim(varf1a(p))//':'//trim(varf2a(p)))
- endif
- endif
- endif
- if (present3) then
- if (present(spval3)) then
- !--- use something other than spval3 if it exists and check consistency
- if (varf3a(p) == spval3) then
- varf3a(p) = recv_varf3(n)
- elseif (recv_varf3(n) /= spval3 .and. varf3a(p) /= recv_varf3(n)) then
- write(nulprt,*) subname//trim(string),astr,'inconsistent linp3 var: ',&
- recv_varf3(n),':',trim(varf1a(p)),':',varf3a(p)
- call oasis_abort(cd_routine=subname//trim(string),cd_message= &
- 'inconsistent linp3 value: '//trim(varf1a(p)))
- endif
- else
- if (varf3a(p) /= recv_varf3(n)) then
- write(nulprt,*) subname//trim(string),astr,'inconsistent linp3 var: ',&
- recv_varf3(n),':',trim(varf1a(p)),':',varf3a(p)
- call oasis_abort(cd_routine=subname//trim(string),cd_message= &
- 'inconsistent linp3 value: '//trim(varf1a(p)))
- endif
- endif
- endif
- endif
- enddo
- if (.not.found) then
- cnt = cnt + 1
- if (cnt > size(varf1a)) then
- allocate(varf1b(size(varf1a)))
- varf1b = varf1a
- deallocate(varf1a)
- if (OASIS_Debug >= 15) then
- write(nulprt,*) subname//trim(string),' resize varf1a ',size(varf1b),cnt+cntr
- call oasis_flush(nulprt)
- endif
- allocate(varf1a(cnt+cntr))
- varf1a(1:size(varf1b)) = varf1b(1:size(varf1b))
- deallocate(varf1b)
- if (present2) then
- allocate(varf2b(size(varf2a)))
- varf2b = varf2a
- deallocate(varf2a)
- if (OASIS_Debug >= 15) then
- write(nulprt,*) subname//trim(string),' resize varf2a ',size(varf2b),cnt+cntr
- call oasis_flush(nulprt)
- endif
- allocate(varf2a(cnt+cntr))
- varf2a(1:size(varf2b)) = varf2b(1:size(varf2b))
- deallocate(varf2b)
- endif
- if (present3) then
- allocate(varf3b(size(varf3a)))
- varf3b = varf3a
- deallocate(varf3a)
- if (OASIS_Debug >= 15) then
- write(nulprt,*) subname//trim(string),' resize varf3a ',size(varf3b),cnt+cntr
- call oasis_flush(nulprt)
- endif
- allocate(varf3a(cnt+cntr))
- varf3a(1:size(varf3b)) = varf3b(1:size(varf3b))
- deallocate(varf3b)
- endif
- endif
- varf1a(cnt) = recv_varf1(n)
- if (present2) varf2a(cnt) = recv_varf2(n)
- if (present3) varf3a(cnt) = recv_varf3(n)
- endif
- enddo ! cntr
- if (local_timers_on) call oasis_timer_stop(trim(string)//'_rl_rootsrch')
- if (cntr > 0) then
- deallocate(recv_varf1)
- if (present2) deallocate(recv_varf2)
- if (present3) deallocate(recv_varf3)
- endif
- endif ! recvid >= 0
- enddo ! maxloops
- !-------------------------------------------------
- !> * Broadcast the list information to all tasks from root
- !-------------------------------------------------
- if (local_timers_on) then
- call oasis_timer_start(trim(string)//'_rl_bcast_barrier')
- if (comm /= MPI_COMM_NULL) &
- call MPI_BARRIER(comm, ierr)
- call oasis_timer_stop(trim(string)//'_rl_bcast_barrier')
- endif
- if (local_timers_on) call oasis_timer_start(trim(string)//'_rl_bcast')
- call oasis_mpi_bcast(cnt,comm,subname//trim(string)//' cnt')
- cntout = cnt
- allocate(lout1(cntout))
- if (commrank == 0) then
- do n = 1,cntout
- lout1(n) = trim(varf1a(n))
- enddo
- endif
- deallocate(varf1a)
- call oasis_mpi_bcast(lout1,comm,subname//trim(string)//' lout1')
- if (present2) then
- allocate(lout2(cntout))
- if (commrank == 0) then
- do n = 1,cntout
- lout2(n) = trim(varf2a(n))
- enddo
- endif
- deallocate(varf2a)
- call oasis_mpi_bcast(lout2,comm,subname//trim(string)//' lout2')
- endif
- if (present3) then
- allocate(lout3(cntout))
- if (commrank == 0) then
- do n = 1,cntout
- lout3(n) = varf3a(n)
- enddo
- endif
- deallocate(varf3a)
- call oasis_mpi_bcast(lout3,comm,subname//trim(string)//' lout3')
- endif
- !--- document
- if (OASIS_debug >= 15) then
- do n = 1,cnt
- if (present2 .and. present3) then
- write(nulprt,*) subname,trim(string),' list: ',n,trim(lout1(n)),' ',trim(lout2(n)),lout3(n)
- elseif (present2) then
- write(nulprt,*) subname,trim(string),' list: ',n,trim(lout1(n)),' ',trim(lout2(n))
- elseif (present3) then
- write(nulprt,*) subname,trim(string),' list: ',n,trim(lout1(n)),lout3(n)
- else
- write(nulprt,*) subname,trim(string),' list: ',n,trim(lout1(n))
- endif
- enddo
- call oasis_flush(nulprt)
- endif
- if (local_timers_on) call oasis_timer_stop (trim(string)//'_rl_bcast')
- call oasis_debug_exit(subname)
- END SUBROUTINE oasis_mpi_reducelists
- !===============================================================================
- !===============================================================================
- END MODULE mod_oasis_mpi
|