aboutsummaryrefslogtreecommitdiffstats
path: root/eval/src/tests/instruction/generic_join
diff options
context:
space:
mode:
authorHåvard Pettersen <havardpe@oath.com>2021-01-14 15:28:22 +0000
committerHåvard Pettersen <havardpe@oath.com>2021-01-14 15:33:53 +0000
commit972e6b8052da175c5db224f19c1480b6238447f6 (patch)
tree252c1d038bd1ec5d82d250fd1e3c885e34c3482a /eval/src/tests/instruction/generic_join
parentc1084399767a5aaf504229ac9c485c696a0832d3 (diff)
forward index when joining a mixed tensor with a dense one
Diffstat (limited to 'eval/src/tests/instruction/generic_join')
-rw-r--r--eval/src/tests/instruction/generic_join/generic_join_test.cpp4
1 files changed, 3 insertions, 1 deletions
diff --git a/eval/src/tests/instruction/generic_join/generic_join_test.cpp b/eval/src/tests/instruction/generic_join/generic_join_test.cpp
index f4046b3d059..7083f4d7eb7 100644
--- a/eval/src/tests/instruction/generic_join/generic_join_test.cpp
+++ b/eval/src/tests/instruction/generic_join/generic_join_test.cpp
@@ -40,7 +40,9 @@ std::vector<Layout> join_layouts = {
{x({"a","b","c"}),y(5)}, {y(5),z({"i","j","k","l"})},
float_cells({x({"a","b","c"}),y(5)}), {y(5),z({"i","j","k","l"})},
{x({"a","b","c"}),y(5)}, float_cells({y(5),z({"i","j","k","l"})}),
- float_cells({x({"a","b","c"}),y(5)}), float_cells({y(5),z({"i","j","k","l"})})
+ float_cells({x({"a","b","c"}),y(5)}), float_cells({y(5),z({"i","j","k","l"})}),
+ {x({"a","b","c"}),y(5)}, float_cells({y(5)}),
+ {y(5)}, float_cells({x({"a","b","c"}),y(5)})
};
bool join_address(const TensorSpec::Address &a, const TensorSpec::Address &b, TensorSpec::Address &addr) {