Merge pull request #1669 from vespa-engine/bratseth/correct-type

kkraune · web-flow · commit b9fe3c2253a0 · 2025-02-24T08:53:07.000+01:00
Use correct value type
diff --git a/billion-scale-image-search/src/test/java/ai/vespa/examples/docproc/DocumentReductionDocProcTest.java b/billion-scale-image-search/src/test/java/ai/vespa/examples/docproc/DocumentReductionDocProcTest.java
@@ -68,7 +68,7 @@ public void test_happy_path()  {
     }
 
     static TensorFieldValue getTensor() {
-        TensorType type = new TensorType.Builder(TensorType.Value.FLOAT).indexed("x", 768).build();
+        TensorType type = new TensorType.Builder(TensorType.Value.BFLOAT16).indexed("x", 768).build();
         IndexedTensor.Builder builder = IndexedTensor.Builder.of(type);
         for (int j = 0; j < 768; j++)
             builder.cell(1.0, j);

Original file line number	Diff line number	Diff line change
`@@ -68,7 +68,7 @@ public void test_happy_path() {`
`68`	`68`	`}`
`69`	`69`
`70`	`70`	`static TensorFieldValue getTensor() {`
`71`		`- TensorType type = new TensorType.Builder(TensorType.Value.FLOAT).indexed("x", 768).build();`
	`71`	`+ TensorType type = new TensorType.Builder(TensorType.Value.BFLOAT16).indexed("x", 768).build();`
`72`	`72`	`IndexedTensor.Builder builder = IndexedTensor.Builder.of(type);`
`73`	`73`	`for (int j = 0; j < 768; j++)`
`74`	`74`	`builder.cell(1.0, j);`