aboutsummaryrefslogtreecommitdiffstats
path: root/eval/src/tests/instruction/generic_join
diff options
context:
space:
mode:
authorArne Juul <arnej@verizonmedia.com>2021-03-01 10:59:49 +0000
committerArne Juul <arnej@verizonmedia.com>2021-03-01 14:04:38 +0000
commit500cd6cd1971b2e6f272a618be95377fc3f8a1b5 (patch)
tree54319f7485e38d0f59c9befcf4ab8fa45d25b22e /eval/src/tests/instruction/generic_join
parent7b7f7c8250492d1f3a2167a4a3f7679375527c60 (diff)
use SmallVector in GenericJoin
Diffstat (limited to 'eval/src/tests/instruction/generic_join')
-rw-r--r--eval/src/tests/instruction/generic_join/generic_join_test.cpp12
1 files changed, 6 insertions, 6 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 994e67934e7..181f44d0f2e 100644
--- a/eval/src/tests/instruction/generic_join/generic_join_test.cpp
+++ b/eval/src/tests/instruction/generic_join/generic_join_test.cpp
@@ -64,9 +64,9 @@ TEST(GenericJoinTest, dense_join_plan_can_be_created) {
auto lhs = ValueType::from_spec("tensor(a{},b[6],c[5],e[3],f[2],g{})");
auto rhs = ValueType::from_spec("tensor(a{},b[6],c[5],d[4],h{})");
auto plan = DenseJoinPlan(lhs, rhs);
- std::vector<size_t> expect_loop = {30,4,6};
- std::vector<size_t> expect_lhs_stride = {6,0,1};
- std::vector<size_t> expect_rhs_stride = {4,1,0};
+ SmallVector<size_t> expect_loop = {30,4,6};
+ SmallVector<size_t> expect_lhs_stride = {6,0,1};
+ SmallVector<size_t> expect_rhs_stride = {4,1,0};
EXPECT_EQ(plan.lhs_size, 180);
EXPECT_EQ(plan.rhs_size, 120);
EXPECT_EQ(plan.out_size, 720);
@@ -80,9 +80,9 @@ TEST(GenericJoinTest, sparse_join_plan_can_be_created) {
auto rhs = ValueType::from_spec("tensor(b[6],c[5],d[4],g{},h{})");
auto plan = SparseJoinPlan(lhs, rhs);
using SRC = SparseJoinPlan::Source;
- std::vector<SRC> expect_sources = {SRC::LHS,SRC::BOTH,SRC::RHS};
- std::vector<size_t> expect_lhs_overlap = {1};
- std::vector<size_t> expect_rhs_overlap = {0};
+ SmallVector<SRC> expect_sources = {SRC::LHS,SRC::BOTH,SRC::RHS};
+ SmallVector<size_t> expect_lhs_overlap = {1};
+ SmallVector<size_t> expect_rhs_overlap = {0};
EXPECT_EQ(plan.sources, expect_sources);
EXPECT_EQ(plan.lhs_overlap, expect_lhs_overlap);
EXPECT_EQ(plan.rhs_overlap, expect_rhs_overlap);