for testing onlylesters/bert-testing

1 files changed, 7 insertions, 0 deletions

diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java
index adb54474812..3211a44fa68 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java
@@ -82,6 +82,13 @@ public class Join extends IntermediateOperation {
             bReducedFunction = new Reduce(b.function().get(), Reduce.Aggregator.sum, bDimensionsToReduce);
         }
 
+        // retain order of inputs
+        if (a == inputs.get(1)) {
+            TensorFunction temp = bReducedFunction;
+            bReducedFunction = aReducedFunction;
+            aReducedFunction = temp;
+        }
+
         return new com.yahoo.tensor.functions.Join(aReducedFunction, bReducedFunction, operator);
     }