added assert for supportsFormat in plugins

2021-11-10 14:37:16 +05:30
parent 936b680f2f
commit 7dd33cd118
12 changed files with 12 additions and 16 deletions
@@ -158,7 +158,7 @@ FlattenConcatRT::configurePlugin(const Dims *inputDims, int32_t nbInputs, const
 }

 bool FlattenConcatRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT {
-    return true;
+    return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR);
 }

 FlattenConcatRTPluginCreator::FlattenConcatRTPluginCreator() {