lars_momentum_op.cu 6.2 KB