arXiv:2505.00136v1 交叉发布类型: 交叉 摘要: Python 是人工智能（AI）领域事实上的软件开发语言。常用的库，如 PyTorch 和 TensorFlow，依赖于其 BLAS 后端内置的并行化，以在 CPU 上实现加速。然而，仅在低级后端应用并行化可能导致性能和扩展性下降。在本文中，我们提出了一种将基于异步运行时模型 HPX 构建的任务级 C++ 代码绑定到高级 Python API 的新方法，使用 pybind11。我们开发了一个并行高斯过程（GP）库作为应用。结果得到的 Python 库 GPRat 结合了常用 GP 库的易用性和异步运行时系统的性能和扩展性。我们在质量-弹簧-阻尼系统上评估了表现，这是一个控制理论中的标准基准，对于不同的回归器（特征）数量。结果显示，在使用 pybind11 绑定异步 HPX 代码时几乎没有绑定开销。与 GPyTorch 和 GPflow 相比，GPRat 在 AMD EPYC 7742 CPU 上高达 64 核的训练中表现出更佳的扩展性。此外，我们的库分别在预测中比 GPyTorch 快 7.63 倍，比 GPflow 快 25.25 倍。如果我们增加特征的数量从八个到 128，我们分别观察到 29.62 倍和 21.19 倍的加速。这些结果展示了在基于 Python 的 AI 应用中使用异步任务的潜力。