Parallelized Real-time Physics Codes for Plasma Control on DIII-D

本文介绍了一种为 DIII-D 等离子体控制系统开发的实时安全多线程库，该库成功地将 TORBEAM 和 STRIDE 物理代码的执行时间分别优化至 20 毫秒和 100 毫秒以下，从而为未来聚变电厂的关键电子回旋波传播和稳定性极限计算提供了支持。

要点

想象一下，像 DIII-D 토카막这样的聚变反应堆就像是一个巨大的、炽热且旋转的电能风暴（等离子体），需要被完美地束缚在一个磁瓶之中。如果这场风暴变得过于狂暴，它就会撞向器壁并摧毁机器。为了保持其安全，一个“等离子体控制系统”（PCS）充当了飞行员的角色，不断进行微小的调整。

然而，风暴的变化速度超过了人类的反应能力。飞行员需要一个超级快速的计算机大脑，能够在眨眼之间预测风暴的行为并提出调整建议。这正是这篇论文的研究所在。

问题所在：“单人工作者”的瓶颈

想象你是一位正在尝试烹饪一场盛大宴会的厨师。你有一份食谱（物理代码）告诉你要如何烹饪食物。但你只有一个厨师（单个计算机处理器核心）在负责所有的切菜、搅拌和烘焙工作。如果食谱太复杂，厨师就会应接不暇，导致食物烧焦，宴会失败。

在聚变领域，这些“食谱”是复杂的物理模拟（例如 TORBEAM 和 STRIDE），它们计算如何加热等离子体或检查等离子体是否即将变得不稳定。传统上，由于这些计算试图仅靠一个“厨师”完成所有工作，因此运行速度太慢，无法满足实时性的要求。

解决方案：“实时安全”的厨师团队

作者构建了一个全新的系统，将那个单人厨师变成了一个步调一致、完美协作的厨师团队。

1. 管理者与工作者： 他们创建了一个特殊的库（一套规则），充当管理者的角色。管理者将小的、独立的任务分配给一组工作者线程（其他计算机核心）。
2. 拒绝混乱，唯有秩序： 在普通的计算机程序中，当你增加更多工作者时，他们可能会产生混乱、等待时间过长，或者如果其中一人出错，会导致整个系统崩溃。作者的系统是“实时安全”的。这就像一支军事单位，每个士兵都清楚地知道何时行动以及何时停止。他们使用一种特殊的“握手”（原子变量）来表达：“我准备好了”、“我完成了”以及“让我们开始下一轮”。
3. 确定性定时： 最重要的一点是，该系统保证会在严格的时间限制内完成工作。无论计算机在处理其他事情时多么忙碌，这个团队都是隔离的，并且始终能按时完成。这至关重要，因为如果计算机耗时过长，等离子体可能已经崩溃了。

他们烹饪的两大主要食谱

该团队利用这个新的“多厨师”系统加速了两个特定的物理代码：

1. TORBEM：激光束引导器

• 功能： 想象一下尝试在黑暗的房间里用激光击中一个移动的小目标。等离子体就是这个房间，而“激光”是用于控制等离子体稳定性的能量束（电子回旋加热）。
• 挑战： 计算机必须计算激光束穿过等离子体的精确路径，以击中正确的位置。
• 结果： 由于每束激光（来自被称为回旋管的不同机器）是独立传播的，新系统允许“工作者”同时计算所有激光束的路径。
• 速度： 他们将计算时间控制在 20 毫秒以内。这足以实时引导激光，保持等离子体的稳定。

2. STRIDE：稳定性检查器

• 功能： 想象一位走钢丝的人。STRIDE 是安全检查员，它不断检查走钢丝的人是否即将跌落。它计算一个“稳定性评分”，以观察等离子体是否即将变得不稳定并发生崩溃。
• 挑战： 这种计算量非常大，通常对于实时使用来说太慢了。
• 技巧： 作者意识到可以将安全检查分解成许多小的、独立的片段（就像检查钢丝的不同部分）。他们将这些片段发送给“工作者”同时求解，然后合并答案。
• 速度： 他们将计算时间缩短到了约 100 毫秒。这足以在灾难发生前向控制系统发出警告。

总结

这篇论文证明，通过构建一个专门的、高度纪律化的计算机处理器团队（一个“实时安全的多线程库”），我们可以足够快地运行复杂的物理模拟，从而在聚变反应堆运行时对其进行实际控制。

• TORBEAM（激光转向）运行在 ~20ms。
• STRIDE（稳定性检查）运行在 ~100ms。

如果没有这个新的“团队协作”系统，这些计算将因速度过慢而无法用于保障聚变反应堆的安全与稳定。这项工作证明，我们可以让聚变反应堆的“飞行员”变得足够聪明，以应对等离子体风暴的极端速度。

技术摘要

技术摘要：用于 DIII-D 等离子体控制的并行化实时物理代码

问题陈述
实时等离子体控制系统（PCS）——例如为 ITER 和 SPARC 等未来聚变反应堆所规划的系统——要求基于物理的代码能够以确定性延迟和高鲁棒性运行。虽然目前已存在用于电子回旋加热（ECH）射线追踪（TORBEAM）和理想磁流体动力学（MHD）稳定性分析（STRIDE）的离线物理代码，但其实时实现往往在单 CPU 核心上遇到性能瓶颈。现有的多线程解决方案（如 OpenMP）并不适用于聚变 PCS 环境，因为它们缺乏硬实时保证，依赖于由具体实现定义的行为，并且由于与内核的交互会引入非确定性的延迟抖动。此外，由于启动成本高且并行粒度不足，GPU 加速对于这些特定代码而言通常并不切实际，因此需要一种稳健的基于 CPU 的并行化策略，以确保微秒级的确定性。

方法论
作者在 DIII-D PCS 上开发并实现了一个自定义的实时安全多线程库，以实现 TORBEAM 和 STRIDE 代码的并行执行。

• 库架构： 该库采用了使用 C11 实现的“管理者-工作者”（Manager-Worker）模式。单个管理者线程（与主 PCS 线程一致）将工作单元分配给多个工作者线程。通信完全通过低层原子变量实现，利用 C11 的获取-释放（acquire-release）内存模型，避免使用像信号量这类与内核交互的高级抽象。
• 同步机制： 系统采用了受 Ada 语言“汇合”（rendezvous）特性启发的双向同步协议。管理者和工作者通过原子标志位（Ready 和 Complete）独立地向对方通知状态。这确保了所有线程在关键部分的开始和结束时都能达到同步点，而不会出现未定义行为。该设计优先考虑确定性而非最大吞吐量，接受线程可能会在等待其他线程追赶时进行空转。
• 硬件环境： 实现运行在具有隔离 CPU 核心的系统上，以防止其他进程和内核中断的干扰。STRIDE 实现利用了商业实时操作系统（RedHawk）来最小化延迟抖动，而 TORBEAM 则在带有核心隔离的标准 CentOS 7 环境下运行。
• 特定代码的并行化：
  • rt-TORBEAM： 对独立回旋管的射线追踪计算进行了并行化处理。代码从高斯束模型简化为单射线追踪，以计算最大吸收位置 $(R, Z)$，随后将其转换为归一化通量坐标 ($\rho$) 用于反射镜转向。
  • rt-STRIDE： 将理想稳定性问题从 Newcomb 方程重新表述为 Riccati 方程。这使得状态转移矩阵（STM）可以通过“打靶”（shooting）法进行计算，即通过将积分区间拆分为若干段。这些区间根据“刚度”指标（在理性面附近步长密度较高）分配给工作者线程，最后由管理者合并这些 STM 以计算最终的稳定性参数 ($\delta W$)。

核心贡献

1. 实时安全多线程库： 开发了一个确定性的、无内核依赖的多线程库，能够在聚变 PCS 硬件上实现安全的并行化，解决了标准库（如 Openable）的局限性。
2. 并行化 rt-TORBEAM： 成功部署了用于 ECH 反射镜转向的并行化射线追踪代码，能够同时处理多个回旋管。
3. 并行化 rt-STRIDE： 实现了一个使用 Riccati 公式和区间拆分技术的并行化理想 MHD 稳定性代码，从而能够实时计算稳定性极限。
4. 实验验证： 证明了这些代码能够在 DIII-D PCS 实际放电过程中，在严格的循环周期约束内运行。

结果

• rt-TORBEAM 性能： 该代码在处理四个转向回旋管时，执行时间始终保持在 20 ms 以下。系统展示了对目标 $\rho$ 值的合理追踪能力（$|\rho_{target} - \rho_{torbeam}| \le 0.05$）。只要存在可用的工作者线程，随着回旋管数量的增加，循环周期保持恒定。在特定边缘情况下观察到循环周期跳升至约 40 ms，这归因于物理代码本身而非线程库。
• rt-STRIDE 性能： 在 72 核系统上，该代码能稳定地在约 100 ms 内计算出 $\delta W$ 稳定性参数。剖析显示，STM 积分本身仅耗时约 20 ms，大部分时间消耗在串行数据预处理（从平衡态到通量坐标的转换）上。
• 追踪与控制： 实验结果表明，虽然物理计算速度足以支持控制，但物理反射镜硬件（移动需约 0.5 s）以及实时平衡态重建的速度限制了整体追踪性能，尤其是在目标值发生快速变化时。

意义与主张
本文认为，开发的并行化多线程库是下一代聚变装置的关键赋能技术，它使基于物理的代码能够满足实时控制的执行时间要求。作者强调，与可能在训练数据之外缺乏鲁棒性的机器学习代理模型不同，这些基于物理的代码能提供“在运行第一天就具备的稳健性能保证”。这项工作证明了通过适当的并行化和确定性线程技术，复杂的物理代码（如 TORBEAM 和 STRIDE）可以集成到 PCS 中，从而为 ECH 沉积和理想稳定性极限提供关键信息。作者指出，该库可应用于其他实时代码，包括垂直稳定性分析和剖面预测，并指出未来的工作重点应放在并行化数据预处理步骤，以进一步降低延迟。