diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-01 11:29:56 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-10 16:35:17 +0000 |
| commit | bfc53031803338d9f0866f88f1d2deffd4928bcc (patch) | |
| tree | b43dbe24df3445639344662e76d1dc37341c2a4f /reference_model/src/generate | |
| parent | 7bf0cb990b55d5738c8dc4291686576654d2d8ab (diff) | |
| download | reference_model-bfc53031803338d9f0866f88f1d2deffd4928bcc.tar.gz | |
Main Compliance testing support for ARGMAX, REDUCE_SUM/MAX/MIN
Add extra tests for FP32 REDUCE_SUM that meet MIN_DOT_PRODUCTS.
Plus improved dot product test generation skip information.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ia8198a9500ddddfc86c5bb84230b9a4edf5ffd50
Diffstat (limited to 'reference_model/src/generate')
| -rw-r--r-- | reference_model/src/generate/generate_dot_product.cc | 44 | ||||
| -rw-r--r-- | reference_model/src/generate/generate_utils.cc | 4 |
2 files changed, 48 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_dot_product.cc b/reference_model/src/generate/generate_dot_product.cc index e6815ad..c1934dd 100644 --- a/reference_model/src/generate/generate_dot_product.cc +++ b/reference_model/src/generate/generate_dot_product.cc @@ -189,6 +189,48 @@ bool generateConv2D(const TosaReference::GenerateConfig& cfg, return false; } } +//---------------------------------------------------------------------------// +// Reduce Sum // +//---------------------------------------------------------------------------// + +bool generateReduceSum(const TosaReference::GenerateConfig& cfg, + TosaReference::IDotProductGenerator& generator, + void* data, + size_t size) +{ + if (cfg.dataType != DType::DType_FP32) + { + WARNING("[Generator][DP][ReduceSum] Only supports FP32."); + return false; + } + if (cfg.inputPos != 0) + { + WARNING("[Generator][DP][ReduceSum] Invalid input tensor slot position to operator."); + return false; + } + if (cfg.dotProductInfo.axis < 0 || static_cast<size_t>(cfg.dotProductInfo.axis) >= cfg.shape.size()) + { + WARNING("[Generator][DP][ReduceSum] Invalid axis %d.", cfg.dotProductInfo.axis); + return false; + } + + float* input = reinterpret_cast<float*>(data); + const int64_t T = TosaReference::numElementsFromShape(cfg.shape); + const uint32_t axis = cfg.dotProductInfo.axis; + + for (int64_t t = 0; t < T; ++t) + { + uint64_t k = t; + for (uint32_t d = cfg.shape.size() - 1; d > axis; --d) + { + k = k / cfg.shape[d]; + } + k = k % cfg.shape[axis]; + + input[t] = generator(static_cast<int32_t>(k)); + } + return true; +} } // namespace namespace TosaReference @@ -210,6 +252,8 @@ bool generateDotProduct(const GenerateConfig& cfg, void* data, size_t size) return generateMatMul(cfg, *generator, data, size); case tosa::Op_CONV2D: return generateConv2D(cfg, *generator, data, size); + case tosa::Op_REDUCE_SUM: + return generateReduceSum(cfg, *generator, data, size); default: WARNING("[Generator][DP] Unsupported operator."); return false; diff --git a/reference_model/src/generate/generate_utils.cc b/reference_model/src/generate/generate_utils.cc index 8ae889a..e410436 100644 --- a/reference_model/src/generate/generate_utils.cc +++ b/reference_model/src/generate/generate_utils.cc @@ -39,6 +39,7 @@ NLOHMANN_JSON_SERIALIZE_ENUM(Op, { { Op::Op_UNKNOWN, "UNKNOWN" }, { Op::Op_ADD, "ADD" }, + { Op::Op_ARGMAX, "ARGMAX" }, { Op::Op_CONV2D, "CONV2D" }, { Op::Op_MATMUL, "MATMUL" }, { Op::Op_MAXIMUM, "MAXIMUM" }, @@ -46,6 +47,9 @@ NLOHMANN_JSON_SERIALIZE_ENUM(Op, { Op::Op_MINIMUM, "MINIMUM" }, { Op::Op_MUL, "MUL" }, { Op::Op_PAD, "PAD" }, + { Op::Op_REDUCE_MAX, "REDUCE_MAX" }, + { Op::Op_REDUCE_MIN, "REDUCE_MIN" }, + { Op::Op_REDUCE_SUM, "REDUCE_SUM" }, { Op::Op_SUB, "SUB" }, }) |