haskell - 如何在运行时设置线程数（避免+RTS -N#）

Question

我希望运行我的并行程序

$ myprogram <args> -n 4 <args>

反而

$ myprogram <args> +RTS -N4 -RTS <args>

主要原因是规范我的程序的参数格式。

我知道它可以做到

$ myprogramwrapper <args> -n 4 <args>
$ cat myprogramwrapper
#!/bin/bash
ARG1=parse args
ARG2=...
NCORES=....
myprogram $ARG1 ... +RTS -N$NCORES

但它很丑。

非常感谢！:)

Answer 1

您可以在运行时使用函数setNumCapabilities。