Missed a logging reference in mpi_context.h

This commit is contained in:
Ryan Hill 2021-05-12 19:39:52 -07:00
parent 3263241baa
commit 06128b9677
2 changed files with 4 additions and 2 deletions

View file

@ -1,6 +1,8 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#define SHARED_PROVIDER_TODO 0
#include "orttraining/core/framework/communication/mpi/mpi_context.h"
#ifndef _WIN32
#include <chrono>
@ -8,8 +10,6 @@
#include <pthread.h>
#endif
#define SHARED_PROVIDER_TODO 0
namespace onnxruntime {
namespace training {

View file

@ -85,7 +85,9 @@ class MPIContext {
int world_size_;
int local_size_;
#if SHARED_PROVIDER_TODO
const logging::Logger& logger_ = logging::LoggingManager::DefaultLogger();
#endif
};
} // namespace training