kernel.cl: Add support for multiple shapes