Add setGPUKernel function to set the current kernel