benchmark some forms of join with number

1 files changed, 18 insertions, 0 deletions

diff --git a/eval/src/tests/tensor/instruction_benchmark/instruction_benchmark.cpp b/eval/src/tests/tensor/instruction_benchmark/instruction_benchmark.cpp
index bcd021b05fb..9ea27ff2f10 100644
--- a/eval/src/tests/tensor/instruction_benchmark/instruction_benchmark.cpp
+++ b/eval/src/tests/tensor/instruction_benchmark/instruction_benchmark.cpp
@@ -705,6 +705,12 @@ TEST(DenseJoin, no_overlap) {
     benchmark_join("dense no overlap multiply", lhs, rhs, operation::Mul::f);
 }
 
+TEST(DenseJoin, multiply_by_number) {
+    auto lhs = make_spec(3.0);
+    auto rhs = make_cube(D::idx("a", 16), D::idx("b", 16), D::idx("c", 16), 2.0);
+    benchmark_join("dense cube multiply by number", lhs, rhs, operation::Mul::f);
+}
+
 //-----------------------------------------------------------------------------
 
 TEST(SparseJoin, small_vectors) {
@@ -743,6 +749,12 @@ TEST(SparseJoin, no_overlap) {
     benchmark_join("sparse no overlap multiply", lhs, rhs, operation::Mul::f);
 }
 
+TEST(SparseJoin, multiply_by_number) {
+    auto lhs = make_spec(3.0);
+    auto rhs = make_cube(D::map("a", 16, 2), D::map("b", 16, 2), D::map("c", 16, 2), 2.0);
+    benchmark_join("sparse multiply by number", lhs, rhs, operation::Mul::f);
+}
+
 //-----------------------------------------------------------------------------
 
 TEST(MixedJoin, full_overlap) {
@@ -763,6 +775,12 @@ TEST(MixedJoin, no_overlap) {
     benchmark_join("mixed no overlap multiply", lhs, rhs, operation::Mul::f);
 }
 
+TEST(MixedJoin, multiply_by_number) {
+    auto lhs = make_spec(3.0);
+    auto rhs = make_cube(D::map("a", 16, 2), D::map("b", 16, 2), D::idx("c", 16), 2.0);
+    benchmark_join("mixed multiply by number", lhs, rhs, operation::Mul::f);
+}
+
 //-----------------------------------------------------------------------------
 
 TEST(ReduceBench, number_reduce) {