saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed macros for graph transformer registration. (#10983)

Browse source
This commit is contained in:
Chandru Ramakrishnan 2022-03-23 14:55:17 -04:00 committed by GitHub
parent de384805cd
commit 07201726ed
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
orttraining/orttraining/core/optimizer/graph_transformer_registry.h
View file

@ -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; \