Pass the right iteration space index to the kernel