summaryrefslogtreecommitdiffstats
path: root/eval/src/apps
diff options
context:
space:
mode:
Diffstat (limited to 'eval/src/apps')
-rw-r--r--eval/src/apps/eval_expr/eval_expr.cpp13
1 files changed, 11 insertions, 2 deletions
diff --git a/eval/src/apps/eval_expr/eval_expr.cpp b/eval/src/apps/eval_expr/eval_expr.cpp
index bc2d2ba8952..5200e37fe5e 100644
--- a/eval/src/apps/eval_expr/eval_expr.cpp
+++ b/eval/src/apps/eval_expr/eval_expr.cpp
@@ -2,6 +2,8 @@
#include <vespa/eval/eval/function.h>
#include <vespa/eval/eval/interpreted_function.h>
+#include <vespa/eval/eval/tensor_spec.h>
+
using namespace vespalib::eval;
@@ -21,7 +23,14 @@ int main(int argc, char **argv) {
InterpretedFunction interpreted(SimpleTensorEngine::ref(), function, NodeTypes());
InterpretedFunction::Context ctx(interpreted);
InterpretedFunction::SimpleParams params({});
- double result = interpreted.eval(ctx, params).as_double();
- fprintf(stdout, "%.32g\n", result);
+ const Value &result = interpreted.eval(ctx, params);
+ if (result.is_double()) {
+ fprintf(stdout, "%.32g\n", result.as_double());
+ } else if (result.is_tensor()) {
+ vespalib::string str = SimpleTensorEngine::ref().to_spec(*result.as_tensor()).to_string();
+ fprintf(stdout, "%s\n", str.c_str());
+ } else {
+ fprintf(stdout, "error\n");
+ }
return 0;
}