Unify init_gpu function in NBNXM