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
@@ -152,7 +152,7 @@ size_t YoloRT::getSerializationSize() const NOEXCEPT {
 }

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

 void YoloRT::serialize(void *buffer) const NOEXCEPT {