< prev index next >

test/jdk/jdk/incubator/vector/Byte256VectorLoadStoreTests.java

Print this page
*** 1,7 ***
  /*
!  * Copyright (c) 2018, 2021, 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.
--- 1,7 ---
  /*
!  * 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.

*** 28,22 ***
   *
   */
  
  // -- This file was mechanically generated: Do not edit! -- //
  
  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 {
--- 28,23 ---
   *
   */
  
  // -- 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.ByteOrder;
  import java.util.List;
  import java.util.function.*;
  
  @Test
  public class Byte256VectorLoadStoreTests extends AbstractVectorLoadStoreTest {

*** 64,11 ***
          } 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));
              }),
--- 65,10 ---

*** 104,11 ***
              withToString("l + speciesl + 1", (int l) -> {
                  return l + SPECIES.length() + 1;
              })
      );
  
!     // Relative to byte[] array.length or ByteBuffer.limit()
      static final List<IntFunction<Integer>> BYTE_INDEX_GENERATORS = List.of(
              withToString("-1", (int l) -> {
                  return -1;
              }),
              withToString("l", (int l) -> {
--- 104,11 ---
              withToString("l + speciesl + 1", (int l) -> {
                  return l + SPECIES.length() + 1;
              })
      );
  
!     // 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) -> {

*** 175,49 ***
                  })).
                  toArray(Object[][]::new);
      }
  
      @DataProvider
!     public Object[][] byteByteBufferProvider() {
          return BYTE_GENERATORS.stream().
!                 flatMap(fa -> BYTE_BUFFER_GENERATORS.stream().
                          flatMap(fb -> BYTE_ORDER_VALUES.stream().map(bo -> {
                              return new Object[]{fa, fb, bo};
                          }))).
                  toArray(Object[][]::new);
      }
  
      @DataProvider
!     public Object[][] byteByteBufferMaskProvider() {
          return BOOLEAN_MASK_GENERATORS.stream().
                  flatMap(fm -> BYTE_GENERATORS.stream().
!                         flatMap(fa -> BYTE_BUFFER_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};
--- 175,30 ---
                  })).
                  toArray(Object[][]::new);
      }
  
      @DataProvider
!     public Object[][] byteMemorySegmentProvider() {
          return BYTE_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[][] byteMemorySegmentMaskProvider() {
          return BOOLEAN_MASK_GENERATORS.stream().
                  flatMap(fm -> BYTE_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[][] byteByteProviderForIOOBE() {
          var f = BYTE_GENERATORS.get(0);
          return BYTE_INDEX_GENERATORS.stream().map(fi -> {
                      return new Object[] {f, fi};

*** 233,32 ***
                      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);
          }
!         return bb.clear();
-     }
- 
-     static byte[] bufferToArray(ByteBuffer bb) {
-         ByteBuffer db = bb;
-         byte[] d = new byte[db.capacity()];
-         db.get(0, d);
-         return d;
      }
  
!     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;
      }
  
  
      interface ToByteF {
          byte apply(int i);
--- 214,20 ---
                      return new Object[] {f, fi, fm};
                  })).
                  toArray(Object[][]::new);
      }
  
!     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 ms;
      }
  
!     static byte[] segmentToArray(MemorySegment ms) {
!         return ms.toArray(ValueLayout.JAVA_BYTE);
      }
  
  
      interface ToByteF {
          byte apply(int i);

*** 294,50 ***
      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);
      }
  
      @DontInline
!     static ByteVector fromByteArray(byte[] a, int i, ByteOrder bo, VectorMask<Byte> m) {
!         return ByteVector.fromByteArray(SPECIES, a, i, bo, m);
      }
  
      @DontInline
!     static void intoByteArray(ByteVector v, byte[] a, int i, ByteOrder bo) {
!         v.intoByteArray(a, i, bo);
      }
  
      @DontInline
!     static void intoByteArray(ByteVector v, byte[] a, int i, ByteOrder bo, VectorMask<Byte> m) {
!         v.intoByteArray(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];
  
--- 263,29 ---
      static void intoArray(ByteVector v, byte[] a, int i, VectorMask<Byte> m) {
          v.intoArray(a, i, m);
      }
  
      @DontInline
!     static ByteVector fromMemorySegment(MemorySegment a, int i, ByteOrder bo) {
!         return ByteVector.fromMemorySegment(SPECIES, a, i, bo);
      }
  
      @DontInline
!     static ByteVector fromMemorySegment(MemorySegment a, int i, ByteOrder bo, VectorMask<Byte> m) {
!         return ByteVector.fromMemorySegment(SPECIES, a, i, bo, m);
      }
  
      @DontInline
!     static void intoMemorySegment(ByteVector v, MemorySegment a, int i, ByteOrder bo) {
!         v.intoMemorySegment(a, i, bo);
      }
  
      @DontInline
!     static void intoMemorySegment(ByteVector v, MemorySegment a, int i, ByteOrder bo, VectorMask<Byte> m) {
!         v.intoMemorySegment(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];
  

*** 504,52 ***
          }
          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());
  
!         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, bo);
!                 av.intoByteBuffer(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");
      }
  
      @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());
  
!         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 = fromByteBuffer(a, i, ByteOrder.nativeOrder());
!                 av.intoByteBuffer(r, i, ByteOrder.nativeOrder());
              }
          }
  
!         int index = fi.apply(a.limit());
!         boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, a.limit());
          try {
!             fromByteBuffer(a, index, ByteOrder.nativeOrder());
              if (shouldFail) {
                  Assert.fail("Failed to throw IndexOutOfBoundsException");
              }
          } catch (IndexOutOfBoundsException e) {
              if (!shouldFail) {
--- 452,49 ---
          }
          Assert.assertEquals(r, mask);
      }
  
  
!     @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 = (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.fromMemorySegment(SPECIES, a, i, bo);
!                 av.intoMemorySegment(r, i, bo);
              }
          }
!         long m = r.mismatch(a);
!         Assert.assertEquals(m, -1, "Segments not equal");
      }
  
      @Test(dataProvider = "byteByteProviderForIOOBE")
!     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 = (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 = fromMemorySegment(a, i, ByteOrder.nativeOrder());
!                 av.intoMemorySegment(r, i, ByteOrder.nativeOrder());
              }
          }
  
!         int index = fi.apply((int) a.byteSize());
!         boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, (int) a.byteSize());
          try {
!             fromMemorySegment(a, index, ByteOrder.nativeOrder());
              if (shouldFail) {
                  Assert.fail("Failed to throw IndexOutOfBoundsException");
              }
          } catch (IndexOutOfBoundsException e) {
              if (!shouldFail) {

*** 557,103 ***
              }
          }
      }
  
      @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());
  
!         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());
              }
          }
  
!         int index = fi.apply(a.limit());
!         boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, a.limit());
          try {
!             ByteVector av = ByteVector.fromByteBuffer(SPECIES, a, 0, ByteOrder.nativeOrder());
!             intoByteBuffer(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) {
          byte[] _a = fa.apply(SPECIES.length());
!         ByteBuffer a = toBuffer(_a, fb);
!         ByteBuffer r = fb.apply(a.limit());
          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, bo, vmask);
!                 av.intoByteBuffer(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);
  
  
!         r = fb.apply(a.limit());
  
          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);
              }
          }
!         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);
      }
  
      @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());
          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 = fromByteBuffer(a, i, ByteOrder.nativeOrder(), vmask);
!                 av.intoByteBuffer(r, i, ByteOrder.nativeOrder());
              }
          }
  
!         int index = fi.apply(a.limit());
!         boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, a.limit(), SPECIES.elementSize() / 8);
          try {
!             fromByteBuffer(a, index, ByteOrder.nativeOrder(), vmask);
              if (shouldFail) {
                  Assert.fail("Failed to throw IndexOutOfBoundsException");
              }
          } catch (IndexOutOfBoundsException e) {
              if (!shouldFail) {
--- 502,94 ---
              }
          }
      }
  
      @Test(dataProvider = "byteByteProviderForIOOBE")
!     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 = (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.fromMemorySegment(SPECIES, a, i, ByteOrder.nativeOrder());
!                 intoMemorySegment(av, r, i, ByteOrder.nativeOrder());
              }
          }
  
!         int index = fi.apply((int) a.byteSize());
!         boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, (int) a.byteSize());
          try {
!             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 = "byteMemorySegmentMaskProvider")
!     static void loadStoreMemorySegmentMask(IntFunction<byte[]> fa,
!                                            IntFunction<MemorySegment> fb,
!                                            IntFunction<boolean[]> fm,
!                                            ByteOrder bo) {
          byte[] _a = fa.apply(SPECIES.length());
!         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 = (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.fromMemorySegment(SPECIES, a, i, bo, vmask);
!                 av.intoMemorySegment(r, i, bo);
              }
          }
!         assertArraysEquals(segmentToArray(r), _a, mask);
  
  
!         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.fromMemorySegment(SPECIES, a, i, bo);
!                 av.intoMemorySegment(r, i, bo, vmask);
              }
          }
!         assertArraysEquals(segmentToArray(r), _a, mask);
      }
  
      @Test(dataProvider = "byteByteMaskProviderForIOOBE")
!     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 = (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 = fromMemorySegment(a, i, ByteOrder.nativeOrder(), vmask);
!                 av.intoMemorySegment(r, i, ByteOrder.nativeOrder());
              }
          }
  
!         int index = fi.apply((int) a.byteSize());
!         boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, (int) a.byteSize(), SPECIES.elementSize() / 8);
          try {
!             fromMemorySegment(a, index, ByteOrder.nativeOrder(), vmask);
              if (shouldFail) {
                  Assert.fail("Failed to throw IndexOutOfBoundsException");
              }
          } catch (IndexOutOfBoundsException e) {
              if (!shouldFail) {

*** 661,248 ***
              }
          }
      }
  
      @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());
          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 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());
              }
          }
  
!         int index = fi.apply(a.length);
!         boolean shouldFail = isIndexOutOfBounds(SPECIES.vectorByteSize(), index, a.length);
          try {
!             ByteVector av = ByteVector.fromByteArray(SPECIES, a, 0, ByteOrder.nativeOrder());
!             intoByteArray(av, r, index, ByteOrder.nativeOrder());
              if (shouldFail) {
                  Assert.fail("Failed to throw IndexOutOfBoundsException");
              }
          } catch (IndexOutOfBoundsException e) {
              if (!shouldFail) {
                  Assert.fail("Unexpected IndexOutOfBoundsException");
              }
          }
      }
  
  
!     @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);
  
!         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(), vmask);
!                 av.intoByteArray(r, i, ByteOrder.nativeOrder());
-             }
-         }
  
!         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");
-             }
-         }
      }
  
-     @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];
--- 597,70 ---
              }
          }
      }
  
      @Test(dataProvider = "byteByteMaskProviderForIOOBE")
!     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 = (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.fromMemorySegment(SPECIES, a, i, ByteOrder.nativeOrder());
!                 intoMemorySegment(av, r, i, ByteOrder.nativeOrder(), vmask);
              }
          }
  
!         int index = fi.apply((int) a.byteSize());
!         boolean shouldFail = isIndexOutOfBoundsForMask(mask, index, (int) a.byteSize(), SPECIES.elementSize() / 8);
          try {
!             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();
  
!         Assert.assertThrows(
!                 UnsupportedOperationException.class,
!                 () -> SPECIES.zero().intoMemorySegment(a, 0, bo)
!         );
  
!         Assert.assertThrows(
!                 UnsupportedOperationException.class,
+                 () -> SPECIES.zero().intoMemorySegment(a, 0, bo, SPECIES.maskAll(true))
+         );
  
!         Assert.assertThrows(
!                 UnsupportedOperationException.class,
!                 () -> SPECIES.zero().intoMemorySegment(a, 0, bo, SPECIES.maskAll(false))
!         );
  
!         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 = "maskProvider")
      static void loadStoreMask(IntFunction<boolean[]> fm) {
          boolean[] a = fm.apply(SPECIES.length());
          boolean[] r = new boolean[a.length];

*** 914,10 ***
--- 672,11 ---
              }
          }
          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 >