Unify gpu_upload_shiftvec(...) function in NBNXM