diff --git a/collects/racket/place/distributed/rmpi.rkt b/collects/racket/place/distributed/rmpi.rkt index e5efb65b07..f879693b03 100644 --- a/collects/racket/place/distributed/rmpi.rkt +++ b/collects/racket/place/distributed/rmpi.rkt @@ -26,6 +26,7 @@ rmpi-partition rmpi-build-default-config rmpi-make-localhost-config + rmpi-make-remote-config rmpi-launch rmpi-finish (struct-out rmpi-comm)) @@ -353,6 +354,12 @@ (list "localhost" (+ start-port i) (string->symbol (format "~a_~a" (symbol->string name) (number->string i))) i))) +(define (rmpi-make-remote-config hostlist cnt start-port name) + (for/list ([i cnt] + [host hostlist]) + (list host (+ start-port i) (string->symbol (format "~a_~a" (symbol->string name) (number->string i))) + i))) + (define (rmpi-alltoall comm outvec) (match-define (rmpi-comm id cnt chs) comm) (define-values (v! vr mkv vcopy)