diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-02-07 10:48:09 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2024-02-08 21:06:29 +0000 |
| commit | 59d8f50f5b3399a6255643aad0e5857e30370761 (patch) | |
| tree | 0139f98187cc15c38bf29fc5022dce514470c6b0 /reference_model | |
| parent | 32d0b5af61d978d9932ac5576b42203e57881168 (diff) | |
| download | reference_model-59d8f50f5b3399a6255643aad0e5857e30370761.tar.gz | |
Fix REDUCE ops to support 8k MAX_RANK
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I365da379f599e5eca1841e6d299b4005a7b0f082
Diffstat (limited to 'reference_model')
| -rw-r--r-- | reference_model/src/ops/reduction.cc | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/reference_model/src/ops/reduction.cc b/reference_model/src/ops/reduction.cc index 2eb764a..c9b85af 100644 --- a/reference_model/src/ops/reduction.cc +++ b/reference_model/src/ops/reduction.cc @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2023, ARM Limited. +// Copyright (c) 2020-2024, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -25,7 +25,7 @@ ReduceNode<Rank, Dtype>::ReduceNode(SubgraphTraverser* sgt_, const Op& op_, Tosa : GraphNode(sgt_, op_, id_) { setRequiredOperands(1, 1); - setRequiredRank(1, 4); + setRequiredRank(1); INIT_ATTRIBUTE(Axis); } @@ -40,6 +40,10 @@ ReduceNode<Rank, Dtype>::~ReduceNode() template <int Rank, TOSA_REF_TYPE Dtype> int ReduceNode<Rank, Dtype>::checkTensorAttributes() { + // Check Tosa Level + auto tosa_level = g_func_config.tosa_level; + LEVEL_CHECK(Rank <= tosa_level.MAX_RANK, "Rank should be smaller than or equal to MAX_RANK"); + if (validateRequiredOperands()) return 1; |