本文探讨了利用 WebGPU 加速客户端零知识证明(ZKP)的方法。WebGPU 是一种可以在多种平台(包括移动设备)上利用 GPU 的技术,通过并行计算和优化内存使用,能够显著提升证明速度。文章详细介绍了 WebGPU 的基本原理、内存层级结构,以及在 WGSL 中实现 NTT(数论变换)的优化过程,最后讨论了将 WebGPU 集成到 ZK 框架中的挑战和未来方向。
本文对Stwo Poseidon AIR进行了基准测试,主要针对WebGPU上proving过程中最耗时的trace生成、trace插值、trace扩展和组合多项式评估等操作进行了性能分析和优化。