From 07201726ed364d8d6abda84d37ad689b5a9289b5 Mon Sep 17 00:00:00 2001
From: Chandru Ramakrishnan <41447659+chandru-r@users.noreply.github.com>
Date: Wed, 23 Mar 2022 14:55:17 -0400
Subject: [PATCH] Fixed macros for graph transformer registration. (#10983)

---
 .../orttraining/core/optimizer/graph_transformer_registry.h   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/orttraining/orttraining/core/optimizer/graph_transformer_registry.h b/orttraining/orttraining/core/optimizer/graph_transformer_registry.h
index afb2100c6a..01727bca79 100644
--- a/orttraining/orttraining/core/optimizer/graph_transformer_registry.h
+++ b/orttraining/orttraining/core/optimizer/graph_transformer_registry.h
@@ -68,7 +68,7 @@ class GraphTransformerRegisterOnce final {
 #define ONNX_REGISTER_EXTERNAL_GRAPH_TRANSFORMER_UNIQ(Counter, name, level, flag) \
   static ONNX_UNUSED onnxruntime::training::GraphTransformerRegisterOnce \
       graph_transformer_register_once##name##Counter(       \
-      #name, [](const std::unordered_set<std::string>& eps) { \
+      #name, [](const InlinedHashSet<std::string_view>& eps) { \
         return std::make_unique<name>(eps);   \
       }, TransformerLevel::level, flag); 
 
@@ -77,7 +77,7 @@ class GraphTransformerRegisterOnce final {
 #define ONNX_REGISTER_EXTERNAL_REWRITE_RULE_UNIQ(Counter, name, level, flag) \
   static ONNX_UNUSED onnxruntime::training::GraphTransformerRegisterOnce \
       graph_transformer_register_once##name##Counter(       \
-      #name, [](const std::unordered_set<std::string>& eps) { \
+      #name, [](const InlinedHashSet<std::string_view>& eps) { \
         auto rule_base_transformer = std::make_unique<RuleBasedGraphTransformer>(#name, eps); \
         ORT_THROW_IF_ERROR(rule_base_transformer->Register(std::make_unique<name>())); \
         return rule_base_transformer;   \