diff --git a/orttraining/orttraining/training_ops/rocm/optimizer/lamb.cc b/orttraining/orttraining/training_ops/rocm/optimizer/lamb.cc index e7ecdfcde0..f45b5b1a85 100644 --- a/orttraining/orttraining/training_ops/rocm/optimizer/lamb.cc +++ b/orttraining/orttraining/training_ops/rocm/optimizer/lamb.cc @@ -651,7 +651,7 @@ Status LambOptimizer::Compute p_ds, p_m1_news, p_m2_news, alpha_, beta_, lambda_, epsilon_, max_norm_clip_, - do_bias_correction_); + do_bias_correction_)); ORT_RETURN_IF_ERROR(launch_lamb_reduction( *this,