Julia:在另一个文件中调用并行循环

问题描述

假设我有这两个文件

test_file.jl

using distributed
function loop(N)
  @distributed for i in 1:N
          println(i)
  end
end

test_file_call.jl

include("test_file.jl")
loop(10)                    

如果我运行 julia -p 2 test_file_call.jl,我希望函数 loop 在不同的处理器上执行,以任意数字打印出 10 个数字。但是,此命令不会呈现任何内容

我不确定我做错了什么?这只是一个简单的循环。是否有可能我在文件 A 中包含一个并行循环,写入另一个包含此循环的文件 B,然后调用 B 在文件 A 中执行并行循环?这两个文件结构是我想要的。这可行吗?

解决方法

问题是您忘记了 @sync 循环,因此它在真正有时间打印任何内容之前就退出了。

因此这应该是:

function loop(N)
  @sync @distributed for i in 1:N
      println(i," ",myid())
  end
end