New Tool Generates Stencil Codes Two Orders of Magnitude Faster on Cerebras WSE Than on GPUs

This post introduces StencilPy, a portable, high-performance code generator for stencil computations, and details its performance on the Cerebras CS-2 system. It highlights that StencilPy reduces the code required for stencil computations on the Cerebras CS-2 by a factor of 7 and achieves significant speedups compared to GPUs (NVIDIA H100) and CPUs (AMD MI210, AMD Genoa). The post elaborates on the Cerebras Wafer-Scale Engine (WSE) architecture, emphasizing its massive parallelism, on-chip memory, and high-bandwidth interconnect as key advantages for stencil computations. It also discusses the Cerebras Software Language (CSL) as the underlying technology enabling these optimizations.