diff options
| author | Jerry Ge <jerry.ge@arm.com> | 2024-06-18 13:26:23 -0700 |
|---|---|---|
| committer | Jerry Ge <jerry.ge@arm.com> | 2024-06-24 16:48:25 -0700 |
| commit | 1c64867fc19cbd5109c3f7a184687a38afb6cd0f (patch) | |
| tree | 041ab4cb477b5dc0091eec562b530ae9954caab3 /reference_model/src/ops/data_layout.cc | |
| parent | 5e1d8b07478d06ddc3bcf1fdd7ad0e5276c1e333 (diff) | |
| download | reference_model-1c64867fc19cbd5109c3f7a184687a38afb6cd0f.tar.gz | |
Update tosa Level_check to include MAX_TENSOR_LIST_SIZE
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Ie729c68a334431d89abb021125dc77819329eb4c
Diffstat (limited to 'reference_model/src/ops/data_layout.cc')
| -rw-r--r-- | reference_model/src/ops/data_layout.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc index 3e3770e..3b8d13a 100644 --- a/reference_model/src/ops/data_layout.cc +++ b/reference_model/src/ops/data_layout.cc @@ -54,6 +54,8 @@ int OpConcat<Rank, Dtype>::checkTensorAttributes() } int32_t num_inputs = inputs.size(); + LEVEL_CHECK(num_inputs <= tosa_level.MAX_TENSOR_LIST_SIZE, + "num_inputs should be smaller than or equal to MAX_TENSOR_LIST_SIZE"); // output and input must be the same types and rank for (int32_t i = 0; i < num_inputs; i++) |