aboutsummaryrefslogtreecommitdiffstats
path: root/eval/src/tests/instruction
diff options
context:
space:
mode:
authorArne Juul <arnej@verizonmedia.com>2021-03-26 13:40:54 +0000
committerArne Juul <arnej@verizonmedia.com>2021-04-06 12:41:23 +0000
commitde080e77a2fd2730e37fcd7c7ae02f9bf36c43c6 (patch)
tree6e1a26138762d0256d8ff28edbee138fe22939f4 /eval/src/tests/instruction
parentaaba78b6f46590c4326fefe205e51c9a026644f1 (diff)
add optimization for join between bfloat16 tensor and number
Diffstat (limited to 'eval/src/tests/instruction')
-rw-r--r--eval/src/tests/instruction/join_with_number/join_with_number_function_test.cpp9
1 files changed, 5 insertions, 4 deletions
diff --git a/eval/src/tests/instruction/join_with_number/join_with_number_function_test.cpp b/eval/src/tests/instruction/join_with_number/join_with_number_function_test.cpp
index a6486de6858..4bdc342b629 100644
--- a/eval/src/tests/instruction/join_with_number/join_with_number_function_test.cpp
+++ b/eval/src/tests/instruction/join_with_number/join_with_number_function_test.cpp
@@ -5,8 +5,8 @@
#include <vespa/eval/eval/test/eval_fixture.h>
#include <vespa/eval/eval/test/gen_spec.h>
#include <vespa/eval/instruction/join_with_number_function.h>
-
#include <vespa/vespalib/util/stringfmt.h>
+#include <vespa/vespalib/util/unwind_message.h>
using namespace vespalib;
using namespace vespalib::eval;
@@ -47,16 +47,17 @@ struct FunInfo {
};
void verify_optimized(const vespalib::string &expr, Primary primary, bool inplace) {
- // fprintf(stderr, "%s\n", expr.c_str());
+ UNWIND_MSG("optimize %s", expr.c_str());
const CellTypeSpace stable_types(CellTypeUtils::list_stable_types(), 2);
FunInfo stable_details{primary, inplace};
TEST_DO(EvalFixture::verify<FunInfo>(expr, {stable_details}, stable_types));
const CellTypeSpace unstable_types(CellTypeUtils::list_unstable_types(), 2);
- TEST_DO(EvalFixture::verify<FunInfo>(expr, {}, unstable_types));
+ FunInfo unstable_details{primary, false};
+ TEST_DO(EvalFixture::verify<FunInfo>(expr, {unstable_details}, unstable_types));
}
void verify_not_optimized(const vespalib::string &expr) {
- // fprintf(stderr, "%s\n", expr.c_str());
+ UNWIND_MSG("not: %s", expr.c_str());
CellTypeSpace all_types(CellTypeUtils::list_types(), 2);
TEST_DO(EvalFixture::verify<FunInfo>(expr, {}, all_types));
}