Make sure context lengths are correct

[lalalune]

For 16x16 and under, our max grid cell count is 256, so we could make the context much shorter in this case.