堆栈 runghc 的多核并行性

Question

我一直在研究script，我希望通过将mapM 换成Async.mapConcurrently 来利用我机器中的多个处理器。

观察到在这种情况下速度没有增加，我想验证runghc 确实可以利用多个内核。

给定一个文件Foo.hs：

import Control.Concurrent

main = print =<< Control.Concurrent.getNumCapabilities

如果我按如下方式编译文件：

stack ghc -- -threaded Foo.hs

然后运行如下：

./Foo

它返回结果1。这是意料之中的，因为没有提供 RTS 选项。改为按如下方式运行：

./Foo +RTS -N

返回数字6，因为我的机器中有6个处理器（同意nproc）。

但是，当我像这样以“解释模式”运行脚本时：

GHCRTS="-N" stack runghc Foo.hs

它会产生以下错误文本：

Running /home/kostmo/.stack/programs/x86_64-linux/ghc-nopie-8.0.2/bin/ghc-pkg --no-user-package-db list --global exited with ExitFailure 1

ghc-pkg: the flag -N requires the program to be built with -threaded

是否可以通过堆栈“脚本”利用多个内核？

Answer 1

感谢您提出这个问题，我认为 stack 应该专门处理 GHCRTS 环境变量，并打开此问题 https://github.com/commercialhaskell/stack/issues/3444 并进行此更改 https://github.com/commercialhaskell/stack/pull/3445

不幸的是，它并没有解决这种情况，因为runghc 本身（ghc）会处理GHCRTS，而且它不是用线程运行时构建的。所以无法使用环境变量解决方案。

我认为应该可以向stack script --compile 提供-with-rtsopts -N 标志，但这似乎不起作用，需要进一步调查。这不适用于runghc，因为它使用解释器。