From ecbd17912ae15ca968816523de9f22018ecf67d2 Mon Sep 17 00:00:00 2001
From: Max Huber <maxh@mailbox.org>
Date: Tue, 12 Sep 2023 11:44:57 +0200
Subject: [PATCH] Fix pooling for large kernel sizes #937

Raise an exception in case an unsupported pooling
operation occurs.

Signed-off-by: Max Huber <maxh@mailbox.org>
---
 onnx2trt_utils.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/onnx2trt_utils.cpp b/onnx2trt_utils.cpp
index b407000..10149de 100644
--- a/onnx2trt_utils.cpp
+++ b/onnx2trt_utils.cpp
@@ -1509,6 +1509,8 @@ NodeImportResult poolingHelper(IImporterContext* ctx, ::ONNX_NAMESPACE::NodeProt
     }
 
     nvinfer1::IPoolingLayer* poolingLayer = ctx->network()->addPoolingNd(*tensorPtr, type, kernel_size);
+    ASSERT(poolingLayer && "Failed to add pooling layer.", ErrorCode::kUNSUPPORTED_NODE);
+
     poolingLayer->setStrideNd(strides);
     // This member is ignored in maxpooling
     poolingLayer->setAverageCountExcludesPadding(exclude_padding);