Update admm_main.cu