[CUDA] More bugs fixed