lars_momentum_op.cu 22.3 KB