diff options
author | Arne H Juul <arnej27959@users.noreply.github.com> | 2023-11-13 16:49:14 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-11-13 16:49:14 +0100 |
commit | 7db2ab348fbb751cd68be69e9460d41f9b1b557e (patch) | |
tree | 2e4896ba7d2b3094639e5824a13d571e38f485ea | |
parent | 0d2cd94a53d87dcfb9db5f5de86d26cb23a0e043 (diff) | |
parent | 250ed42d709a8d4bddecb5ba35d9e2a0991f1f01 (diff) |
Merge pull request #29323 from vespa-engine/havardpe/extend-conformance-test
extend conformance test with map_subspaces
-rw-r--r-- | eval/src/apps/tensor_conformance/generate.cpp | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/eval/src/apps/tensor_conformance/generate.cpp b/eval/src/apps/tensor_conformance/generate.cpp index 0ebfb772451..1c68a94a852 100644 --- a/eval/src/apps/tensor_conformance/generate.cpp +++ b/eval/src/apps/tensor_conformance/generate.cpp @@ -236,6 +236,28 @@ void generate_map(TestBuilder &dst) { //----------------------------------------------------------------------------- +void generate_map_subspaces(TestBuilder &dst) { + auto my_seq = Seq({-128, -43, 85, 127}); + auto scalar = GenSpec(7.0); + auto sparse = GenSpec().from_desc("x8_1").seq(my_seq); + auto mixed = GenSpec().from_desc("x4_1y4").seq(my_seq); + auto dense = GenSpec().from_desc("y4").seq(my_seq); + vespalib::string map_a("map_subspaces(a,f(a)(a*3+2))"); + vespalib::string unpack_a("map_subspaces(a,f(a)(tensor<int8>(y[8])(bit(a,7-y%8))))"); + vespalib::string unpack_y4("map_subspaces(a,f(a)(tensor<int8>(y[32])(bit(a{y:(y/8)},7-y%8))))"); + vespalib::string pack_y4("map_subspaces(a,f(a)(a{y:0}+a{y:1}-a{y:2}+a{y:3}))"); + generate(map_a, scalar, dst); + generate(map_a, sparse, dst); + generate(unpack_a, scalar, dst); + generate(unpack_a, sparse, dst); + generate(unpack_y4, mixed, dst); + generate(unpack_y4, dense, dst); + generate(pack_y4, mixed, dst); + generate(pack_y4, dense, dst); +} + +//----------------------------------------------------------------------------- + void generate_join_expr(const vespalib::string &expr, const Sequence &seq, TestBuilder &dst) { for (const auto &layouts: join_layouts) { GenSpec a = GenSpec::from_desc(layouts.first).seq(seq); @@ -560,6 +582,7 @@ Generator::generate(TestBuilder &dst) generate_inject(dst); generate_reduce(dst); generate_map(dst); + generate_map_subspaces(dst); generate_join(dst); generate_merge(dst); generate_concat(dst); |