From 37f4f74308ba856c866e589c99dae15dbdd62f7c Mon Sep 17 00:00:00 2001
From: "Tang, Cheng" <souptc@gmail.com>
Date: Wed, 22 Apr 2020 21:35:41 -0700
Subject: [PATCH] expose training session so the training app could register
 custom kernel and transformers (#3642)

Co-authored-by: Cheng Tang <chenta@microsoft.com>
---
 orttraining/orttraining/models/runner/training_runner.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/orttraining/orttraining/models/runner/training_runner.h b/orttraining/orttraining/models/runner/training_runner.h
index 23f7672c2e..94d972edca 100644
--- a/orttraining/orttraining/models/runner/training_runner.h
+++ b/orttraining/orttraining/models/runner/training_runner.h
@@ -169,6 +169,7 @@ class TrainingRunner {
   common::Status ResetLossScaler();
 
   size_t GetRound() const { return round_; }
+  TrainingSession& GetSession() { return session_; }
 
  private:
   Status TrainingLoop(IDataLoader& training_data_loader, IDataLoader* test_data_loader);