From 149c037a4a2e1085ac70af2d577284b827da618a Mon Sep 17 00:00:00 2001
From: Ryan Hill <ryanhill@microsoft.com>
Date: Mon, 10 May 2021 16:18:56 -0700
Subject: [PATCH] For orttraining

---
 orttraining/orttraining/models/bert/main.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/orttraining/orttraining/models/bert/main.cc b/orttraining/orttraining/models/bert/main.cc
index a71e6c2481..1c6b02cdc5 100644
--- a/orttraining/orttraining/models/bert/main.cc
+++ b/orttraining/orttraining/models/bert/main.cc
@@ -616,6 +616,7 @@ void setup_training_params(BertParameters& params) {
         0,
         true,
         0,
+        nullptr,
         nullptr};
 
     if (params.gpu_mem_limit_in_gb > 0) {