From 01743d0308a805a1f2ddb945125b970c21fa8b3a Mon Sep 17 00:00:00 2001 From: Serge Panev Date: Tue, 4 Aug 2020 11:40:32 -0700 Subject: [PATCH] Add CHECK_EQ Signed-off-by: Serge Panev --- src/operator/subgraph/tensorrt/tensorrt-inl.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/operator/subgraph/tensorrt/tensorrt-inl.h b/src/operator/subgraph/tensorrt/tensorrt-inl.h index b35a1715000e..5c9ba355cd61 100644 --- a/src/operator/subgraph/tensorrt/tensorrt-inl.h +++ b/src/operator/subgraph/tensorrt/tensorrt-inl.h @@ -275,10 +275,11 @@ class TensorrtProperty : public SubgraphProperty { NDArray **in_aux_ptr = g.GetAttr("in_aux"); in_args_dict.clear(); in_aux_dict.clear(); - // we trust the Python API, len(in_arg_names) == len(in_args_ptr) + CHECK_EQ(in_arg_names.size(), in_args_ptr.size()); for (unsigned i = 0; i < in_arg_names.size(); ++i) { in_args_dict[in_arg_names[i]] = in_args_ptr[i]; } + CHECK_EQ(in_aux_names.size(), in_aux_ptr.size()); for (unsigned i = 0; i < in_aux_names.size(); ++i) { in_aux_dict[in_aux_names[i]] = in_aux_ptr[i]; }