adamw_kernel.cu 11.1 KB