Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test default and panama impls return the same result

This commit is contained in:
ChrisHegarty 2024-12-18 15:32:21 +00:00
parent fbf112a727
commit 62cd45ba54
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
lucene/core/src/test/org/apache/lucene/internal/vectorization/TestVectorUtilSupport.java
View File

@ -20,6 +20,7 @@ import com.carrotsearch.randomizedtesting.annotations.ParametersFactory;
import java.util.Arrays;
import java.util.function.ToDoubleFunction;
import java.util.function.ToIntFunction;
import java.util.function.ToLongFunction;
import java.util.stream.IntStream;
public class TestVectorUtilSupport extends BaseVectorizationTestCase {
@ -133,6 +134,27 @@ public class TestVectorUtilSupport extends BaseVectorizationTestCase {
PANAMA_PROVIDER.getVectorUtilSupport().int4DotProduct(a, false, pack(b), true));
}
public void testInt4BitDotProduct() {
var binaryQuantized = new byte[size];
var int4Quantized = new byte[size * 4];
random().nextBytes(binaryQuantized);
random().nextBytes(int4Quantized);
assertLongReturningProviders(p -> p.int4BitDotProduct(int4Quantized, binaryQuantized));
}
public void testInt4BitDotProductBoundaries() {
var binaryQuantized = new byte[size];
var int4Quantized = new byte[size * 4];
Arrays.fill(binaryQuantized, Byte.MAX_VALUE);
Arrays.fill(int4Quantized, Byte.MAX_VALUE);
assertLongReturningProviders(p -> p.int4BitDotProduct(int4Quantized, binaryQuantized));
Arrays.fill(binaryQuantized, Byte.MIN_VALUE);
Arrays.fill(int4Quantized, Byte.MIN_VALUE);
assertLongReturningProviders(p -> p.int4BitDotProduct(int4Quantized, binaryQuantized));
}
static byte[] pack(byte[] unpacked) {
int len = (unpacked.length + 1) / 2;
var packed = new byte[len];
@ -154,4 +176,10 @@ public class TestVectorUtilSupport extends BaseVectorizationTestCase {
func.applyAsInt(LUCENE_PROVIDER.getVectorUtilSupport()),
func.applyAsInt(PANAMA_PROVIDER.getVectorUtilSupport()));
}
private void assertLongReturningProviders(ToLongFunction<VectorUtilSupport> func) {
assertEquals(
func.applyAsLong(LUCENE_PROVIDER.getVectorUtilSupport()),
func.applyAsLong(PANAMA_PROVIDER.getVectorUtilSupport()));
}
}