Skip to content

Commit 8afab22

Browse files
committed
fix: Fix plugin registration macro
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 0a2405f commit 8afab22

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

core/plugins/impl/interpolate_plugin.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -388,9 +388,9 @@ const nvinfer1::PluginFieldCollection* InterpolatePluginCreator::getFieldNames()
388388
return nullptr;
389389
}
390390

391-
REGISTER_TRTORCH_PLUGIN(InterpolatePluginCreator);
391+
REGISTER_TORCHTRT_PLUGIN(InterpolatePluginCreator);
392392

393393
} // namespace impl
394394
} // namespace plugins
395395
} // namespace core
396-
} // namespace trtorch
396+
} // namespace trtorch

core/plugins/impl/normalize_plugin.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -264,9 +264,9 @@ const nvinfer1::PluginFieldCollection* NormalizePluginCreator::getFieldNames() n
264264
return nullptr;
265265
}
266266

267-
REGISTER_TRTORCH_PLUGIN(NormalizePluginCreator);
267+
REGISTER_TORCHTRT_PLUGIN(NormalizePluginCreator);
268268

269269
} // namespace impl
270270
} // namespace plugins
271271
} // namespace core
272-
} // namespace trtorch
272+
} // namespace trtorch

core/plugins/plugins.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ class PluginRegistrar {
2020
T instance{};
2121
};
2222

23-
#define REGISTER_TRTORCH_PLUGIN(name) \
23+
#define REGISTER_TORCHTRT_PLUGIN(name) \
2424
static PluginRegistrar<name> pluginRegistrar##name {}
2525

2626
} // namespace impl

0 commit comments

Comments
 (0)