aboutsummaryrefslogtreecommitdiffstats
path: root/eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp
diff options
context:
space:
mode:
authorHåvard Pettersen <havardpe@yahooinc.com>2022-06-07 10:01:15 +0000
committerHåvard Pettersen <havardpe@yahooinc.com>2022-06-07 10:01:15 +0000
commit05019f25b0d748dd34bbcfb2dc839611e09d96f0 (patch)
tree753ee21e96270fe03d10b8ab7e05d962c4d11798 /eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp
parent38e71d4979792c42b0d163268ad1335cf3176b37 (diff)
full reduce with COUNT aggregator is cell count
Diffstat (limited to 'eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp')
-rw-r--r--eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp b/eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp
index fe531785278..d7298db4b68 100644
--- a/eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp
+++ b/eval/src/tests/instruction/generic_reduce/generic_reduce_test.cpp
@@ -75,7 +75,7 @@ void test_generic_reduce_with(const ValueBuilderFactory &factory) {
auto input = layout.cpy().cells(ct);
if (input.bad_scalar()) continue;
SCOPED_TRACE(fmt("tensor type: %s, num_cells: %zu", input.gen().type().c_str(), input.gen().cells().size()));
- for (Aggr aggr: {Aggr::SUM, Aggr::AVG, Aggr::MIN, Aggr::MAX}) {
+ for (Aggr aggr: {Aggr::SUM, Aggr::AVG, Aggr::MIN, Aggr::MAX, Aggr::COUNT}) {
SCOPED_TRACE(fmt("aggregator: %s", AggrNames::name_of(aggr)->c_str()));
auto t = layout.type();
for (const auto & dim: t.dimensions()) {