Wasi: WebGPU —— 为 WebAssembly 引入 GPU 计算能力的系统接口提案

背景

WebAssembly（Wasm）凭借其便携性、安全性和沙箱隔离特性，已成为跨平台计算的重要基石。然而，长期以来，Wasm 在图形处理单元（GPU）计算方面的支持主要依赖于 Web 环境下的 WebGPU API。随着 Wasm 应用场景从浏览器扩展到服务器、边缘计算、移动设备乃至桌面端，开发者迫切需要一个独立于 Web 和 JavaScript 环境的标准化 GPU 访问接口。

在此背景下，Mendy Berger 和 Sean Isom 提出了 wasi:webgpu 提案。该提案旨在通过 WebAssembly System Interface (WASI) 规范，将 GPU 计算能力带入非 Web 环境的 Wasm 运行时中。目前该提案处于 Phase 2（第二阶段），支持 Linux、Windows、macOS、Android 以及 Web 平台。这一举措标志着 Wasm 生态正从单纯的“Web 替代方案”向“通用系统级运行时”演进，特别是在高性能计算、AI 推理和图形渲染领域。

核心内容

wasi:webgpu 是一个 WASI 提案，专门用于在 WebAssembly 中提供 GPU 访问能力。其核心设计哲学是充分利用 Wasm 的优势，同时尽可能复用现有的行业标准。

1. 设计目标与用例

该提案旨在将 Wasm 的便携性、安全性和沙箱机制延伸至 GPU 计算领域。主要应用场景包括但不限于：

• 服务端图形流媒体：在服务器端进行高效的图形编码和流式传输。
• 科学计算与模拟：利用 GPU 加速复杂的物理模拟、气象预测或分子动力学计算。
• AI/ML 推理与训练：在边缘设备或服务器端运行机器学习模型的推理甚至训练任务。
• 图像与视频处理：实时处理高分辨率图像和视频流。
• 数据可视化与渲染：在客户端或服务器端进行大规模数据的可视化呈现。

需要注意的是，显示到屏幕或窗口管理 API（如直接操作操作系统窗口）不在 wasi:webgpu 的范围内。这部分功能正在其他项目中活跃开发（例如 wasi-gfx），未来可能会成为独立的 WASI 提案。

2. 基于 WebGPU 规范

wasi:webgpu 直接基于官方的 WebGPU 规范构建。这意味着开发者如果熟悉 Web 环境下的 WebGPU API，可以较快地迁移到 Wasm 环境中。

3. 与 Web 环境的差异

尽管基于 WebGPU 规范，wasi:webgpu 在部分设计上与 Web 环境下的规范存在偏差。这些偏差主要源于 WebGPU 规范中隐含的“运行在 Web 或 JavaScript 环境中”的假设。例如：

• 内存模型差异：Web 环境通常通过 JavaScript 对象引用内存，而 Wasm 使用线性内存（Linear Memory）。
• 生命周期管理：Web 环境有垃圾回收（GC），而 Wasm 需要更明确的生命周期控制。

对于每一个偏离 WebGPU 规范的地方，提案都要求提供清晰的解释文档，以确保 API 行为的可预测性和一致性。

4. API 设计与权衡

完整的 API 文档定义在 imports.md 文件中。详细的 API 设计讨论主要集中在 .wit.md（WebAssembly Interface Types）文件中，该文件规定了接口的具体形态。

在 API 设计过程中，提案团队进行了深入的权衡讨论（Trade-offs），例如：

• 同步 vs 异步：如何在保持 Wasm 同步执行特性的同时，处理 GPU 的异步计算任务。
• 错误处理：如何在不依赖 JavaScript 异常机制的情况下，优雅地处理 GPU 驱动错误或资源不足。
• 安全性边界：如何确保 Wasm 模块无法访问未授权的 GPU 资源或泄露敏感数据。

5. 替代方案考虑

提案中探讨了其他可能的实现路径，但最终选择了基于 WebGPU 规范进行适配的方案。主要理由包括：

• 生态兼容性：WebGPU 已成为行业标准，基于它构建可以减少学习成本。
• 实现复用：现有的 WebGPU 实现（如 Chrome、Firefox、Safari 的底层实现）可以为 WASI 提供基础支持。
• 避免碎片化：如果创建全新的 GPU API，会导致 Wasm 生态的碎片化，增加实现者的负担。

关键要点

• 标准化 GPU 访问：wasi:webgpu 为 WebAssembly 提供了标准化的 GPU 接口，填补了非 Web 环境下 Wasm 缺乏原生 GPU 支持的空白。
• 跨平台支持：提案明确支持 Linux、Windows、macOS、Android 和 Web，体现了 Wasm “一次编写，到处运行”的愿景。
• 复用 WebGPU 标准：API 设计紧密跟随官方 WebGPU 规范，仅在必要时针对非 JS 环境进行调整，降低了迁移门槛。
• 聚焦计算，剥离显示：明确将“GPU 计算”与“屏幕显示/窗口管理”分离。显示功能由其他提案（如 wasi-gfx）负责，使 wasi:webgpu 更专注于计算密集型任务。
• 处于 Phase 2：目前提案已进入第二阶段，意味着核心设计已基本确定，正在收集利益相关者的反馈并准备进入实施阶段。
• 需要实现者承诺：在进入 Phase 3 之前，提案需要列出已表达实施兴趣的实现者（Implementers），以确保提案的可行性和落地能力。

意义与影响

wasi:webgpu 的提出是 WebAssembly 生态发展中的一个重要里程碑，其影响深远：

1. 解锁高性能计算新场景： 通过赋予 Wasm 原生 GPU 访问能力，开发者可以在服务器端、边缘设备甚至嵌入式系统中运行原本只能在浏览器或特定 GPU 驱动中运行的计算密集型应用。这将极大推动 AI 推理、科学模拟和实时图形处理在 Wasm 生态中的普及。

2. 增强安全性与隔离性： 在云原生和边缘计算场景中，wasi:webgpu 允许在不信任的环境中运行 GPU 代码，同时保持 Wasm 的沙箱安全特性。这对于多租户云服务、第三方插件系统以及需要严格资源隔离的应用至关重要。

3. 促进跨语言与跨平台开发： 由于 WASI 是语言无关的，wasi:webgpu 使得 Rust、C/C++、AssemblyScript 等语言编写的 Wasm 模块能够共享同一套 GPU 接口。这降低了 GPU 编程的门槛，促进了不同语言生态之间的互操作性。

4. 推动 WebGPU 标准的泛化： 通过将 WebGPU 规范适配到 WASI 环境，wasi:webgpu 反过来也促进了 WebGPU 标准本身的完善和通用化。这种“从 Web 到系统”的反馈循环有助于建立一个更健壮、更通用的图形计算标准。

5. 为未来图形 API 奠定基础： 虽然 wasi:webgpu 不包含窗口管理，但它为后续的 wasi-gfx 等提案奠定了计算基础。未来，完整的图形栈（计算+渲染+显示）可能在 WASI 中形成统一标准，彻底改变 Web 和原生应用的开发模式。

总之，wasi:webgpu 不仅是一个技术提案，更是 Wasm 向通用计算平台演进的关键一步。它预示着未来我们可能在任何地方、以任何语言、安全高效地利用 GPU 算力，而不再受限于特定的浏览器或操作系统环境。