如何根据CPU核心缩放线程？

Question

我想用 Java 中的多线程解决数学问题。我的数学问题可以分为多个工作单元，我想在多个线程中解决。

我不想让固定数量的线程在其上工作，而是让线程数量与 CPU 核心数量相匹配。我的问题是，我无法在互联网上找到简单的教程。我发现的都是具有固定线程的示例。

这怎么能做到呢？你能提供例子吗？

Answer 1

您可以使用静态运行时方法来确定 Java 虚拟机可用的进程数，可用处理器。一旦确定了可用处理器的数量，就可以创建该数量的线程并相应地分配工作。

更新：为了进一步澄清，线程只是 Java 中的一个对象，因此您可以像创建任何其他对象一样创建它。因此，假设您调用上述方法并发现它返回 2 个处理器。惊人的。现在，您可以创建一个循环来生成新线程，并拆分该线程的工作，然后触发该线程。下面是一些伪代码来演示我的意思：

int processors = Runtime.getRuntime().availableProcessors();
for(int i=0; i < processors; i++) {
  Thread yourThread = new AThreadYouCreated();
  // You may need to pass in parameters depending on what work you are doing and how you setup your thread.
  yourThread.start();
}

有关创建自己的线程的更多信息，前往本教程。另外，您可能需要查看线程池来创建线程。

Answer 2

您可能还想查看 java.util.concurrent 框架来了解这些内容。
像这样：

ExecutorService e = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
// Do work using something like either
e.execute(new Runnable() {
        public void run() {
            // do one task
        }
    });

或者

    Future<String> future = pool.submit(new Callable<String>() {
        public String call() throws Exception {
            return null;
        }
    });
    future.get();  // Will block till result available

这比处理你自己的线程池等要好得多。

Answer 3

选项 1：

newWorkStealingPool

public static ExecutorService newWorkStealingPool()

使用所有可用处理器作为目标并行级别创建工作窃取线程池。

使用此 API，您无需将核心数传递给 ExecutorService。

此 API 的实现来自 grepcode

/**
     * Creates a work-stealing thread pool using all
     * {@link Runtime#availableProcessors available processors}
     * as its target parallelism level.
     * @return the newly created thread pool
     * @see #newWorkStealingPool(int)
     * @since 1.8
     */
    public static ExecutorService newWorkStealingPool() {
        return new ForkJoinPool
            (Runtime.getRuntime().availableProcessors(),
             ForkJoinPool.defaultForkJoinWorkerThreadFactory,
             null, true);
    }

选项 2：

newFixedThreadPool 来自 Executors 或 其他 newXXX 构造函数，返回 ExecutorService

public static ExecutorService newFixedThreadPool(int nThreads)

将 nThreads 替换为 Runtime.getRuntime().availableProcessors()

选项 3： >

ThreadPoolExecutor

public ThreadPoolExecutor(int corePoolSize,
                      int maximumPoolSize,
                      long keepAliveTime,
                      TimeUnit unit,
                      BlockingQueue<Runnable> workQueue)

通过Runtime.getRuntime().availableProcessors() 作为 maximumPoolSize 的参数。

Answer 4

Doug Lea（并发包的作者）有一篇可能相关的论文：
http://gee.cs.oswego.edu/dl/papers/fj.pdf

Fork Join 框架已添加到 Java SE 7 中。以下是更多参考资料：

http://www.oracle.com/technetwork /articles/java/fork-join-422606.html

Answer 5

标准方法是 Runtime.getRuntime().availableProcessors() 方法。
在大多数标准 CPU 上，您将在此处返回最佳线程数（不是实际的 CPU 核心数）。因此，这就是您正在寻找的。

示例：

ExecutorService service = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());

不要忘记像这样关闭执行程序服务（否则您的程序将不会退出）：

service.shutdown();

这里只是一个快速概述如何设置基于未来的 MT 代码（题外话，用于说明）：

CompletionService<YourCallableImplementor> completionService = 
    new ExecutorCompletionService<YourCallableImplementor>(service);
    ArrayList<Future<YourCallableImplementor>> futures = new ArrayList<Future<YourCallableImplementor>>();
    for (String computeMe : elementsToCompute) {
        futures.add(completionService.submit(new YourCallableImplementor(computeMe)));
    }

然后您需要跟踪您期望有多少结果并像这样检索它们：

try {
  int received = 0;
  while (received < elementsToCompute.size()) {
     Future<YourCallableImplementor> resultFuture = completionService.take(); 
     YourCallableImplementor result = resultFuture.get();
     received++; 
  }
} finally {
  service.shutdown();
}

Answer 6

在 Runtime 类上，有一个名为 availableProcessors() 的方法。您可以使用它来计算您有多少个 CPU。由于您的程序受 CPU 限制，因此您可能希望每个可用 CPU （最多）有一个线程。