[CUDA] Added more checks