Add macro mp_async

Author: oschulz

src/onprocs.jl

@@ -33,6 +33,31 @@ end
 export @onprocs
 
 
+"""
+    @mp_async expr
+
+Run `expr` asynchronously on a worker process.
+
+Compatible with `@sync`.
+
+Equivalent to `Distributed.@spawn expr` on Julia <= v1.2, equivalent to
+`Distributed.@spawn :any expr` on Julia >= v1.3.
+"""
+macro mp_async(expr)
+    # Code taken from Distributed.@spawn:
+    thunk = esc(:(()->($expr)))
+    var = esc(Base.sync_varname)
+    quote
+        local ref = Distributed.spawn_somewhere($thunk)
+        if $(Expr(:isdefined, var))
+            push!($var, ref)
+        end
+        ref
+    end
+end
+export @mp_async
+
+
 function mtjulia_exe()
     if Sys.islinux()
         joinpath(@__DIR__, "..", "bin", "mtjulia.sh")

test/test_onprocs.jl

@@ -30,6 +30,20 @@ using Distributed
         end) == ref_result
     end
 
+    @testset "macro mp_async" begin
+        @test begin
+            n = 128
+            A = Vector{Future}(undef, n)
+            @sync for i in 1:n
+                A[i] = @mp_async begin
+                    @assert myid() != 1
+                    log(i)
+                end
+            end
+            fetch.(A) == log.(1:n)
+        end
+    end
+
     @testset "mtjulia_exe" begin
         if Sys.islinux()
             @test fetch(@spawnat first(workers()) nthreads()) > 1