Cerebras (CA) Interview Question

Implementing 2D prefix sum, multithreaded