Add support for 32x32 and smaller CUs