That's right, I found it: https://docs.cupy.dev/en/stable/tutorial/kernel.html#kernel-...