< prev index next > test/jdk/jdk/incubator/vector/Byte256VectorLoadStoreTests.java
Print this page
/*
- * Copyright (c) 2018, 2021, Oracle and/or its affiliates. All rights reserved.
+ * Copyright (c) 2018, 2022, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
*/
// -- This file was mechanically generated: Do not edit! -- //
+ import jdk.incubator.foreign.MemorySegment;
+ import jdk.incubator.foreign.ResourceScope;
+ import jdk.incubator.foreign.ValueLayout;
import jdk.incubator.vector.ByteVector;
import jdk.incubator.vector.VectorMask;
import jdk.incubator.vector.VectorSpecies;
import jdk.incubator.vector.VectorShuffle;
import jdk.internal.vm.annotation.DontInline;
import org.testng.Assert;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
- import java.nio.ByteBuffer;
import java.nio.ByteOrder;
- import java.nio.ReadOnlyBufferException;
import java.util.List;
import java.util.function.*;
@Test
public class Byte256VectorLoadStoreTests extends AbstractVectorLoadStoreTest {
} catch (AssertionError e) {
Assert.assertEquals(r[i], mask[i % SPECIES.length()] ? a[i] : (byte) 0, "at index #" + i);
}
}
-
static final List<IntFunction<byte[]>> BYTE_GENERATORS = List.of(
withToString("byte[i * 5]", (int s) -> {
return fill(s * BUFFER_REPS,
i -> (byte)(i * 5));
}),
withToString("l + speciesl + 1", (int l) -> {
return l + SPECIES.length() + 1;
})
);
- // Relative to byte[] array.length or ByteBuffer.limit()
+ // Relative to byte[] array.length or MemorySegment.byteSize()
static final List<IntFunction<Integer>> BYTE_INDEX_GENERATORS = List.of(
withToString("-1", (int l) -> {
return -1;
}),
withToString("l", (int l) -> {
})).
toArray(Object[][]::new);
}
@DataProvider
- public Object[][] byteByteBufferProvider() {
+ public Object[][] byteMemorySegmentProvider() {
return BYTE_GENERATORS.stream().
- flatMap(fa -> BYTE_BUFFER_GENERATORS.stream().
+ flatMap(fa -> MEMORY_SEGMENT_GENERATORS.stream().
flatMap(fb -> BYTE_ORDER_VALUES.stream().map(bo -> {
return new Object[]{fa, fb, bo};
}))).
toArray(Object[][]::new);
}
@DataProvider
- public Object[][] byteByteBufferMaskProvider() {
+ public Object[][] byteMemorySegmentMaskProvider() {
return BOOLEAN_MASK_GENERATORS.stream().
flatMap(fm -> BYTE_GENERATORS.stream().
- flatMap(fa -> BYTE_BUFFER_GENERATORS.stream().
+ flatMap(fa -> MEMORY_SEGMENT_GENERATORS.stream().
flatMap(fb -> BYTE_ORDER_VALUES.stream().map(bo -> {
return new Object[]{fa, fb, fm, bo};
})))).
toArray(Object[][]::new);
}
- @DataProvider
- public Object[][] byteByteArrayProvider() {
- return BYTE_GENERATORS.stream().
- flatMap(fa -> BYTE_ORDER_VALUES.stream().map(bo -> {
- return new Object[]{fa, bo};
- })).
- toArray(Object[][]::new);
- }
-
- @DataProvider
- public Object[][] byteByteArrayMaskProvider() {
- return BOOLEAN_MASK_GENERATORS.stream().
- flatMap(fm -> BYTE_GENERATORS.stream().
- flatMap(fa -> BYTE_ORDER_VALUES.stream().map(bo -> {
- return new Object[]{fa, fm, bo};
- }))).
- toArray(Object[][]::new);
- }
-
@DataProvider
public Object[][] byteByteProviderForIOOBE() {
var f = BYTE_GENERATORS.get(0);
return BYTE_INDEX_GENERATORS.stream().map(fi -> {
return new Object[] {f, fi};
return new Object[] {f, fi, fm};
})).
toArray(Object[][]::new);
}
- static ByteBuffer toBuffer(byte[] a, IntFunction<ByteBuffer> fb) {
- ByteBuffer bb = fb.apply(a.length * SPECIES.elementSize() / 8);
- for (byte v : a) {
- bb.put(v);
+ static MemorySegment toSegment(byte[] a, IntFunction<MemorySegment> fb) {
+ MemorySegment ms = fb.apply(a.length * SPECIES.elementSize() / 8);
+ for (int i = 0; i < a.length; i++) {
+ ms.set(ValueLayout.JAVA_BYTE, i * SPECIES.elementSize() / 8 , a[i]);
}
- return bb.clear();
- }
-
- static byte[] bufferToArray(ByteBuffer bb) {
- ByteBuffer db = bb;
- byte[] d = new byte[db.capacity()];
- db.get(0, d);
- return d;
+ return ms;
}
- static byte[] toByteArray(byte[] a, IntFunction<byte[]> fb, ByteOrder bo) {
- byte[] b = fb.apply(a.length * SPECIES.elementSize() / 8);
- ByteBuffer bb = ByteBuffer.wrap(b, 0, b.length).order(bo);
- for (byte v : a) {
- bb.put(v);
- }
- return b;
+ static byte[] segmentToArray(MemorySegment ms) {
+ return ms.toArray(ValueLayout.JAVA_BYTE);
}
interface ToByteF {
byte apply(int i);
static void intoArray(ByteVector v, byte[] a, int i, VectorMask<Byte> m) {
v.intoArray(a, i, m);
}
@DontInline
- static ByteVector fromByteArray(byte[] a, int i, ByteOrder bo) {
- return ByteVector.fromByteArray(SPECIES, a, i, bo);
+ static ByteVector fromMemorySegment(MemorySegment a, int i, ByteOrder bo) {
+ return ByteVector.fromMemorySegment(SPECIES, a, i, bo);
}
@DontInline
- static ByteVector fromByteArray(byte[] a, int i, ByteOrder bo, VectorMask<Byte> m) {
- return ByteVector.fromByteArray(SPECIES, a, i, bo, m);
+ static ByteVector fromMemorySegment(MemorySegment a, int i, ByteOrder bo, VectorMask<Byte> m) {
+ return ByteVector.fromMemorySegment(SPECIES, a, i, bo, m);
}
@DontInline
- static void intoByteArray(ByteVector v, byte[] a, int i, ByteOrder bo) {
- v.intoByteArray(a, i, bo);
+ static void intoMemorySegment(ByteVector v, MemorySegment a, int i, ByteOrder bo) {
+ v.intoMemorySegment(a, i, bo);
}
@DontInline
- static void intoByteArray(ByteVector v, byte[] a, int i, ByteOrder bo, VectorMask<Byte> m) {
- v.intoByteArray(a, i, bo, m);
+ static void intoMemorySegment(ByteVector v, MemorySegment a, int i, ByteOrder bo, VectorMask<Byte> m) {
+ v.intoMemorySegment(a, i, bo, m);
}
- @DontInline
- static ByteVector fromByteBuffer(ByteBuffer a, int i, ByteOrder bo) {
- return ByteVector.fromByteBuffer(SPECIES, a, i, bo);
- }
-
- @DontInline
- static ByteVector fromByteBuffer(ByteBuffer a, int i, ByteOrder bo, VectorMask<Byte> m) {
- return ByteVector.fromByteBuffer(SPECIES, a, i, bo, m);
- }
-
- @DontInline
- static void intoByteBuffer(ByteVector v, ByteBuffer a, int i, ByteOrder bo) {
- v.intoByteBuffer(a, i, bo);
- }
-
- @DontInline
- static void intoByteBuffer(ByteVector v, ByteBuffer a, int i, ByteOrder bo, VectorMask<Byte> m) {
- v.intoByteBuffer(a, i, bo, m);
- }
-
-
@Test(dataProvider = "byteProvider")
static void loadStoreArray(IntFunction<byte[]> fa) {
byte[] a = fa.apply(SPECIES.length());
byte[] r = new byte[a.length];
}
Assert.assertEquals(r, mask);
}
- @Test(dataProvider = "byteByteBufferProvider")
- static void loadStoreByteBuffer(IntFunction<byte[]> fa,
- IntFunction<ByteBuffer> fb,
- ByteOrder bo) {
- ByteBuffer a = toBuffer(fa.apply(SPECIES.length()), fb);
- ByteBuffer r = fb.apply(a.limit());
+ @Test(dataProvider = "byteMemorySegmentProvider")
+ static void loadStoreMemorySegment(IntFunction<byte[]> fa,
+ IntFunction<MemorySegment> fb,
+ ByteOrder bo) {
+ MemorySegment a = toSegment(fa.apply(SPECIES.length()), fb);
+ MemorySegment r = fb.apply((int) a.byteSize());
- int l = a.limit();
+ int l = (int) a.byteSize();
int s = SPECIES.vectorByteSize();
for (int ic = 0; ic < INVOC_COUNT; ic++) {
for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, i, bo);
- av.intoByteBuffer(r, i, bo);
+ ByteVector av = ByteVector.fromMemorySegment(SPECIES, a, i, bo);
+ av.intoMemorySegment(r, i, bo);
}
}
- Assert.assertEquals(a.position(), 0, "Input buffer position changed");
- Assert.assertEquals(a.limit(), l, "Input buffer limit changed");
- Assert.assertEquals(r.position(), 0, "Result buffer position changed");
- Assert.assertEquals(r.limit(), l, "Result buffer limit changed");
- Assert.assertEquals(r, a, "Buffers not equal");
+ long m = r.mismatch(a);
+ Assert.assertEquals(m, -1, "Segments not equal");
}
@Test(dataProvider = "byteByteProviderForIOOBE")
- static void loadByteBufferIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi) {
- ByteBuffer a = toBuffer(fa.apply(SPECIES.length()), ByteBuffer::allocateDirect);
- ByteBuffer r = ByteBuffer.allocateDirect(a.limit());
+ static void loadMemorySegmentIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi) {
+ MemorySegment a = toSegment(fa.apply(SPECIES.length()), i -> MemorySegment.allocateNative(i, ResourceScope.newImplicitScope()));
+ MemorySegment r = MemorySegment.allocateNative(a.byteSize(), ResourceScope.newImplicitScope());
- int l = a.limit();
+ int l = (int) a.byteSize();
int s = SPECIES.vectorByteSize();
for (int ic = 0; ic < INVOC_COUNT; ic++) {
for (int i = 0; i < l; i += s) {
- ByteVector av = fromByteBuffer(a, i, ByteOrder.nativeOrder());
- av.intoByteBuffer(r, i, ByteOrder.nativeOrder());
+ ByteVector av = fromMemorySegment(a, i, ByteOrder.nativeOrder());
+ av.intoMemorySegment(r, i, ByteOrder.nativeOrder());
}
}
- int index = fi.apply(a.limit());
- boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, a.limit());
+ int index = fi.apply((int) a.byteSize());
+ boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, (int) a.byteSize());
try {
- fromByteBuffer(a, index, ByteOrder.nativeOrder());
+ fromMemorySegment(a, index, ByteOrder.nativeOrder());
if (shouldFail) {
Assert.fail("Failed to throw IndexOutOfBoundsException");
}
} catch (IndexOutOfBoundsException e) {
if (!shouldFail) {
}
}
}
@Test(dataProvider = "byteByteProviderForIOOBE")
- static void storeByteBufferIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi) {
- ByteBuffer a = toBuffer(fa.apply(SPECIES.length()), ByteBuffer::allocateDirect);
- ByteBuffer r = ByteBuffer.allocateDirect(a.limit());
+ static void storeMemorySegmentIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi) {
+ MemorySegment a = toSegment(fa.apply(SPECIES.length()), i -> MemorySegment.allocateNative(i, ResourceScope.newImplicitScope()));
+ MemorySegment r = MemorySegment.allocateNative(a.byteSize(), ResourceScope.newImplicitScope());
- int l = a.limit();
+ int l = (int) a.byteSize();
int s = SPECIES.vectorByteSize();
for (int ic = 0; ic < INVOC_COUNT; ic++) {
for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, i, ByteOrder.nativeOrder());
- intoByteBuffer(av, r, i, ByteOrder.nativeOrder());
+ ByteVector av = ByteVector.fromMemorySegment(SPECIES, a, i, ByteOrder.nativeOrder());
+ intoMemorySegment(av, r, i, ByteOrder.nativeOrder());
}
}
- int index = fi.apply(a.limit());
- boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, a.limit());
+ int index = fi.apply((int) a.byteSize());
+ boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, (int) a.byteSize());
try {
- ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, 0, ByteOrder.nativeOrder());
- intoByteBuffer(av, r, index, ByteOrder.nativeOrder());
+ ByteVector av = ByteVector.fromMemorySegment(SPECIES, a, 0, ByteOrder.nativeOrder());
+ intoMemorySegment(av, r, index, ByteOrder.nativeOrder());
if (shouldFail) {
Assert.fail("Failed to throw IndexOutOfBoundsException");
}
} catch (IndexOutOfBoundsException e) {
if (!shouldFail) {
Assert.fail("Unexpected IndexOutOfBoundsException");
}
}
}
-
- @Test(dataProvider = "byteByteBufferMaskProvider")
- static void loadStoreByteBufferMask(IntFunction<byte[]> fa,
- IntFunction<ByteBuffer> fb,
- IntFunction<boolean[]> fm,
- ByteOrder bo) {
+ @Test(dataProvider = "byteMemorySegmentMaskProvider")
+ static void loadStoreMemorySegmentMask(IntFunction<byte[]> fa,
+ IntFunction<MemorySegment> fb,
+ IntFunction<boolean[]> fm,
+ ByteOrder bo) {
byte[] _a = fa.apply(SPECIES.length());
- ByteBuffer a = toBuffer(_a, fb);
- ByteBuffer r = fb.apply(a.limit());
+ MemorySegment a = toSegment(_a, fb);
+ MemorySegment r = fb.apply((int) a.byteSize());
boolean[] mask = fm.apply(SPECIES.length());
VectorMask<Byte> vmask = VectorMask.fromValues(SPECIES, mask);
- int l = a.limit();
+ int l = (int) a.byteSize();
int s = SPECIES.vectorByteSize();
for (int ic = 0; ic < INVOC_COUNT; ic++) {
for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, i, bo, vmask);
- av.intoByteBuffer(r, i, bo);
+ ByteVector av = ByteVector.fromMemorySegment(SPECIES, a, i, bo, vmask);
+ av.intoMemorySegment(r, i, bo);
}
}
- Assert.assertEquals(a.position(), 0, "Input buffer position changed");
- Assert.assertEquals(a.limit(), l, "Input buffer limit changed");
- Assert.assertEquals(r.position(), 0, "Result buffer position changed");
- Assert.assertEquals(r.limit(), l, "Result buffer limit changed");
- assertArraysEquals(bufferToArray(r), _a, mask);
+ assertArraysEquals(segmentToArray(r), _a, mask);
- r = fb.apply(a.limit());
+ r = fb.apply((int) a.byteSize());
for (int ic = 0; ic < INVOC_COUNT; ic++) {
for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, i, bo);
- av.intoByteBuffer(r, i, bo, vmask);
+ ByteVector av = ByteVector.fromMemorySegment(SPECIES, a, i, bo);
+ av.intoMemorySegment(r, i, bo, vmask);
}
}
- Assert.assertEquals(a.position(), 0, "Input buffer position changed");
- Assert.assertEquals(a.limit(), l, "Input buffer limit changed");
- Assert.assertEquals(r.position(), 0, "Result buffer position changed");
- Assert.assertEquals(r.limit(), l, "Result buffer limit changed");
- assertArraysEquals(bufferToArray(r), _a, mask);
+ assertArraysEquals(segmentToArray(r), _a, mask);
}
@Test(dataProvider = "byteByteMaskProviderForIOOBE")
- static void loadByteBufferMaskIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi, IntFunction<boolean[]> fm) {
- ByteBuffer a = toBuffer(fa.apply(SPECIES.length()), ByteBuffer::allocateDirect);
- ByteBuffer r = ByteBuffer.allocateDirect(a.limit());
+ static void loadMemorySegmentMaskIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi, IntFunction<boolean[]> fm) {
+ MemorySegment a = toSegment(fa.apply(SPECIES.length()), i -> MemorySegment.allocateNative(i, ResourceScope.newImplicitScope()));
+ MemorySegment r = MemorySegment.allocateNative(a.byteSize(), ResourceScope.newImplicitScope());
boolean[] mask = fm.apply(SPECIES.length());
VectorMask<Byte> vmask = VectorMask.fromValues(SPECIES, mask);
- int l = a.limit();
+ int l = (int) a.byteSize();
int s = SPECIES.vectorByteSize();
for (int ic = 0; ic < INVOC_COUNT; ic++) {
for (int i = 0; i < l; i += s) {
- ByteVector av = fromByteBuffer(a, i, ByteOrder.nativeOrder(), vmask);
- av.intoByteBuffer(r, i, ByteOrder.nativeOrder());
+ ByteVector av = fromMemorySegment(a, i, ByteOrder.nativeOrder(), vmask);
+ av.intoMemorySegment(r, i, ByteOrder.nativeOrder());
}
}
- int index = fi.apply(a.limit());
- boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, a.limit(), SPECIES.elementSize() / 8);
+ int index = fi.apply((int) a.byteSize());
+ boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, (int) a.byteSize(), SPECIES.elementSize() / 8);
try {
- fromByteBuffer(a, index, ByteOrder.nativeOrder(), vmask);
+ fromMemorySegment(a, index, ByteOrder.nativeOrder(), vmask);
if (shouldFail) {
Assert.fail("Failed to throw IndexOutOfBoundsException");
}
} catch (IndexOutOfBoundsException e) {
if (!shouldFail) {
}
}
}
@Test(dataProvider = "byteByteMaskProviderForIOOBE")
- static void storeByteBufferMaskIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi, IntFunction<boolean[]> fm) {
- ByteBuffer a = toBuffer(fa.apply(SPECIES.length()), ByteBuffer::allocateDirect);
- ByteBuffer r = ByteBuffer.allocateDirect(a.limit());
+ static void storeMemorySegmentMaskIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi, IntFunction<boolean[]> fm) {
+ MemorySegment a = toSegment(fa.apply(SPECIES.length()), i -> MemorySegment.allocateNative(i, ResourceScope.newImplicitScope()));
+ MemorySegment r = MemorySegment.allocateNative(a.byteSize(), ResourceScope.newImplicitScope());
boolean[] mask = fm.apply(SPECIES.length());
VectorMask<Byte> vmask = VectorMask.fromValues(SPECIES, mask);
- int l = a.limit();
- int s = SPECIES.vectorByteSize();
-
- for (int ic = 0; ic < INVOC_COUNT; ic++) {
- for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, i, ByteOrder.nativeOrder());
- intoByteBuffer(av, r, i, ByteOrder.nativeOrder(), vmask);
- }
- }
-
- int index = fi.apply(a.limit());
- boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, a.limit(), SPECIES.elementSize() / 8);
- try {
- ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, 0, ByteOrder.nativeOrder());
- intoByteBuffer(av, a, index, ByteOrder.nativeOrder(), vmask);
- if (shouldFail) {
- Assert.fail("Failed to throw IndexOutOfBoundsException");
- }
- } catch (IndexOutOfBoundsException e) {
- if (!shouldFail) {
- Assert.fail("Unexpected IndexOutOfBoundsException");
- }
- }
- }
-
-
- @Test(dataProvider = "byteByteBufferProvider")
- static void loadStoreReadonlyByteBuffer(IntFunction<byte[]> fa,
- IntFunction<ByteBuffer> fb,
- ByteOrder bo) {
- ByteBuffer a = toBuffer(fa.apply(SPECIES.length()), fb).asReadOnlyBuffer();
-
- try {
- SPECIES.zero().intoByteBuffer(a, 0, bo);
- Assert.fail("ReadOnlyBufferException expected");
- } catch (ReadOnlyBufferException e) {
- }
-
- try {
- SPECIES.zero().intoByteBuffer(a, 0, bo, SPECIES.maskAll(true));
- Assert.fail("ReadOnlyBufferException expected");
- } catch (ReadOnlyBufferException e) {
- }
-
- try {
- SPECIES.zero().intoByteBuffer(a, 0, bo, SPECIES.maskAll(false));
- Assert.fail("ReadOnlyBufferException expected");
- } catch (ReadOnlyBufferException e) {
- }
-
- try {
- VectorMask<Byte> m = SPECIES.shuffleFromOp(i -> i % 2 == 0 ? 1 : -1)
- .laneIsValid();
- SPECIES.zero().intoByteBuffer(a, 0, bo, m);
- Assert.fail("ReadOnlyBufferException expected");
- } catch (ReadOnlyBufferException e) {
- }
- }
-
-
- @Test(dataProvider = "byteByteArrayProvider")
- static void loadStoreByteArray(IntFunction<byte[]> fa,
- ByteOrder bo) {
- byte[] a = toByteArray(fa.apply(SPECIES.length()), byte[]::new, bo);
- byte[] r = new byte[a.length];
-
- int s = SPECIES.vectorByteSize();
- int l = a.length;
-
- for (int ic = 0; ic < INVOC_COUNT; ic++) {
- for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteArray(SPECIES, a, i, bo);
- av.intoByteArray(r, i, bo);
- }
- }
- Assert.assertEquals(r, a, "Byte arrays not equal");
- }
-
- @Test(dataProvider = "byteByteProviderForIOOBE")
- static void loadByteArrayIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi) {
- byte[] a = toByteArray(fa.apply(SPECIES.length()), byte[]::new, ByteOrder.nativeOrder());
- byte[] r = new byte[a.length];
-
- int s = SPECIES.vectorByteSize();
- int l = a.length;
-
- for (int ic = 0; ic < INVOC_COUNT; ic++) {
- for (int i = 0; i < l; i += s) {
- ByteVector av = fromByteArray(a, i, ByteOrder.nativeOrder());
- av.intoByteArray(r, i, ByteOrder.nativeOrder());
- }
- }
-
- int index = fi.apply(a.length);
- boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, a.length);
- try {
- fromByteArray(a, index, ByteOrder.nativeOrder());
- if (shouldFail) {
- Assert.fail("Failed to throw IndexOutOfBoundsException");
- }
- } catch (IndexOutOfBoundsException e) {
- if (!shouldFail) {
- Assert.fail("Unexpected IndexOutOfBoundsException");
- }
- }
- }
-
- @Test(dataProvider = "byteByteProviderForIOOBE")
- static void storeByteArrayIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi) {
- byte[] a = toByteArray(fa.apply(SPECIES.length()), byte[]::new, ByteOrder.nativeOrder());
- byte[] r = new byte[a.length];
-
+ int l = (int) a.byteSize();
int s = SPECIES.vectorByteSize();
- int l = a.length;
for (int ic = 0; ic < INVOC_COUNT; ic++) {
for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteArray(SPECIES, a, i, ByteOrder.nativeOrder());
- intoByteArray(av, r, i, ByteOrder.nativeOrder());
+ ByteVector av = ByteVector.fromMemorySegment(SPECIES, a, i, ByteOrder.nativeOrder());
+ intoMemorySegment(av, r, i, ByteOrder.nativeOrder(), vmask);
}
}
- int index = fi.apply(a.length);
- boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, a.length);
+ int index = fi.apply((int) a.byteSize());
+ boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, (int) a.byteSize(), SPECIES.elementSize() / 8);
try {
- ByteVector av = ByteVector.fromByteArray(SPECIES, a, 0, ByteOrder.nativeOrder());
- intoByteArray(av, r, index, ByteOrder.nativeOrder());
+ ByteVector av = ByteVector.fromMemorySegment(SPECIES, a, 0, ByteOrder.nativeOrder());
+ intoMemorySegment(av, a, index, ByteOrder.nativeOrder(), vmask);
if (shouldFail) {
Assert.fail("Failed to throw IndexOutOfBoundsException");
}
} catch (IndexOutOfBoundsException e) {
if (!shouldFail) {
Assert.fail("Unexpected IndexOutOfBoundsException");
}
}
}
+ @Test(dataProvider = "byteMemorySegmentProvider")
+ static void loadStoreReadonlyMemorySegment(IntFunction<byte[]> fa,
+ IntFunction<MemorySegment> fb,
+ ByteOrder bo) {
+ MemorySegment a = toSegment(fa.apply(SPECIES.length()), fb).asReadOnly();
- @Test(dataProvider = "byteByteArrayMaskProvider")
- static void loadStoreByteArrayMask(IntFunction<byte[]> fa,
- IntFunction<boolean[]> fm,
- ByteOrder bo) {
- byte[] a = toByteArray(fa.apply(SPECIES.length()), byte[]::new, bo);
- byte[] r = new byte[a.length];
- boolean[] mask = fm.apply(SPECIES.length());
- VectorMask<Byte> vmask = VectorMask.fromValues(SPECIES, mask);
-
- int s = SPECIES.vectorByteSize();
- int l = a.length;
-
- for (int ic = 0; ic < INVOC_COUNT; ic++) {
- for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteArray(SPECIES, a, i, bo, vmask);
- av.intoByteArray(r, i, bo);
- }
- }
- assertArraysEquals(r, a, mask);
-
-
- r = new byte[a.length];
-
- for (int ic = 0; ic < INVOC_COUNT; ic++) {
- for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteArray(SPECIES, a, i, bo);
- av.intoByteArray(r, i, bo, vmask);
- }
- }
- assertArraysEquals(r, a, mask);
- }
-
- @Test(dataProvider = "byteByteMaskProviderForIOOBE")
- static void loadByteArrayMaskIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi, IntFunction<boolean[]> fm) {
- byte[] a = toByteArray(fa.apply(SPECIES.length()), byte[]::new, ByteOrder.nativeOrder());
- byte[] r = new byte[a.length];
- boolean[] mask = fm.apply(SPECIES.length());
- VectorMask<Byte> vmask = VectorMask.fromValues(SPECIES, mask);
+ Assert.assertThrows(
+ UnsupportedOperationException.class,
+ () -> SPECIES.zero().intoMemorySegment(a, 0, bo)
+ );
- int s = SPECIES.vectorByteSize();
- int l = a.length;
+ Assert.assertThrows(
+ UnsupportedOperationException.class,
+ () -> SPECIES.zero().intoMemorySegment(a, 0, bo, SPECIES.maskAll(true))
+ );
- for (int ic = 0; ic < INVOC_COUNT; ic++) {
- for (int i = 0; i < l; i += s) {
- ByteVector av = fromByteArray(a, i, ByteOrder.nativeOrder(), vmask);
- av.intoByteArray(r, i, ByteOrder.nativeOrder());
- }
- }
+ Assert.assertThrows(
+ UnsupportedOperationException.class,
+ () -> SPECIES.zero().intoMemorySegment(a, 0, bo, SPECIES.maskAll(false))
+ );
- int index = fi.apply(a.length);
- boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, a.length, SPECIES.elementSize() / 8);
- try {
- fromByteArray(a, index, ByteOrder.nativeOrder(), vmask);
- if (shouldFail) {
- Assert.fail("Failed to throw IndexOutOfBoundsException");
- }
- } catch (IndexOutOfBoundsException e) {
- if (!shouldFail) {
- Assert.fail("Unexpected IndexOutOfBoundsException");
- }
- }
+ VectorMask<Byte> m = SPECIES.shuffleFromOp(i -> i % 2 == 0 ? 1 : -1)
+ .laneIsValid();
+ Assert.assertThrows(
+ UnsupportedOperationException.class,
+ () -> SPECIES.zero().intoMemorySegment(a, 0, bo, m)
+ );
}
- @Test(dataProvider = "byteByteMaskProviderForIOOBE")
- static void storeByteArrayMaskIOOBE(IntFunction<byte[]> fa, IntFunction<Integer> fi, IntFunction<boolean[]> fm) {
- byte[] a = toByteArray(fa.apply(SPECIES.length()), byte[]::new, ByteOrder.nativeOrder());
- byte[] r = new byte[a.length];
- boolean[] mask = fm.apply(SPECIES.length());
- VectorMask<Byte> vmask = VectorMask.fromValues(SPECIES, mask);
-
- int s = SPECIES.vectorByteSize();
- int l = a.length;
-
- for (int ic = 0; ic < INVOC_COUNT; ic++) {
- for (int i = 0; i < l; i += s) {
- ByteVector av = ByteVector.fromByteArray(SPECIES, a, i, ByteOrder.nativeOrder());
- intoByteArray(av, r, i, ByteOrder.nativeOrder(), vmask);
- }
- }
-
- int index = fi.apply(a.length);
- boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, a.length, SPECIES.elementSize() / 8);
- try {
- ByteVector av = ByteVector.fromByteArray(SPECIES, a, 0, ByteOrder.nativeOrder());
- intoByteArray(av, a, index, ByteOrder.nativeOrder(), vmask);
- if (shouldFail) {
- Assert.fail("Failed to throw IndexOutOfBoundsException");
- }
- } catch (IndexOutOfBoundsException e) {
- if (!shouldFail) {
- Assert.fail("Unexpected IndexOutOfBoundsException");
- }
- }
- }
@Test(dataProvider = "maskProvider")
static void loadStoreMask(IntFunction<boolean[]> fm) {
boolean[] a = fm.apply(SPECIES.length());
boolean[] r = new boolean[a.length];
}
}
Assert.assertEquals(r, a);
}
+
@Test
static void loadStoreShuffle() {
IntUnaryOperator fn = a -> a + 5;
for (int ic = 0; ic < INVOC_COUNT; ic++) {
var shuffle = VectorShuffle.fromOp(SPECIES, fn);
< prev index next >