[CUDA] Reverted from 1D->2D to plain 1D