Cerebras (CA) Interview Question

A question about resource allocation and contexts. I think it's a graph coloring problem.