[arcane,utils] Ajoute support accélérateur pour 'MultiArray2'.

Ajoute support pour choisir l'allocateur mémoire et créé une classe
'MultiArray2SmallSpan' pour obtenir une vue sur l'instance.

Author: grospelliergilles

arcane/src/arcane/utils/MultiArray2.h

@@ -67,6 +67,7 @@ class MultiArray2
  public:
 
   MultiArray2() = default;
+  // TODO: Rendre accessible uniquement à UniqueMultiArray2 ou SharedMultiArray2
   explicit MultiArray2(ConstArrayView<Int32> sizes)
   {
     _resize(sizes);
@@ -96,6 +97,17 @@ class MultiArray2
   , m_sizes(aview.m_sizes)
   {
   }
+  explicit MultiArray2(const MemoryAllocationOptions& allocation_options)
+  : m_buffer(allocation_options)
+  , m_indexes(allocation_options)
+  , m_sizes(allocation_options)
+  {}
+  // TODO: Rendre accessible uniquement à UniqueMultiArray2 ou SharedMultiArray2
+  MultiArray2(const MemoryAllocationOptions& allocation_options, ConstArrayView<Int32> sizes)
+  : MultiArray2(allocation_options)
+  {
+    _resize(sizes);
+  }
 
  public:
 
@@ -111,23 +123,15 @@ class MultiArray2
  public:
 
   //! Nombre total d'éléments
-  Integer totalNbElement() const { return m_buffer.size(); }
+  Int32 totalNbElement() const { return m_buffer.size(); }
+
   //! Supprime les éléments du tableau.
   void clear()
   {
     m_buffer.clear();
     m_indexes.clear();
     m_sizes.clear();
   }
-  //! Clone le tableau
-  MultiArray2<DataType> clone()
-  {
-    MultiArray2<DataType> new_array;
-    new_array.m_buffer = m_buffer.clone();
-    new_array.m_indexes = m_indexes.clone();
-    new_array.m_sizes = m_sizes.clone();
-    return new_array;
-  }
   //! Remplit les éléments du tableau avec la valeur \a v
   void fill(const DataType& v)
   {
@@ -137,7 +141,7 @@ class MultiArray2
   {
     return m_buffer[m_indexes[i] + j];
   }
-  const DataType& at(Integer i, Integer j) const
+  ConstReferenceType at(Integer i, Integer j) const
   {
     return m_buffer[m_indexes[i] + j];
   }
@@ -178,6 +182,24 @@ class MultiArray2
     return ConstMultiArray2View<DataType>(m_buffer, m_indexes, m_sizes);
   }
 
+  //! Vue modifiable du tableau
+  MultiArray2SmallSpan<DataType> span()
+  {
+    return { m_buffer.smallSpan(), m_indexes, m_sizes };
+  }
+
+  //! Vue constante du tableau
+  MultiArray2SmallSpan<const DataType> span() const
+  {
+    return { m_buffer, m_indexes, m_sizes };
+  }
+
+  //! Vue constante du tableau
+  MultiArray2SmallSpan<const DataType> constSpan() const
+  {
+    return { m_buffer.constSmallSpan(), m_indexes, m_sizes };
+  }
+
   //! Vue du tableau sous forme de tableau 1D
   ArrayView<DataType> viewAsArray()
   {
@@ -233,7 +255,7 @@ class MultiArray2
 
     Integer old_size1 = m_indexes.size();
 
-    SharedArray<DataType> new_buffer(total_size);
+    SharedArray<DataType> new_buffer(m_buffer.allocationOptions(), total_size);
 
     // Recopie dans le nouveau tableau les valeurs de l'ancien.
     if (old_size1 > size1)
@@ -278,7 +300,7 @@ class MultiArray2
 
  private:
 
-  //! Valeurs
+  //! Tableau des Valeurs
   SharedArray<DataType> m_buffer;
   //! Tableau des indices dans \a m_buffer du premièr élément de la deuxième dimension
   SharedArray<Int32> m_indexes;
@@ -361,6 +383,16 @@ class UniqueMultiArray2
   explicit UniqueMultiArray2(ConstArrayView<Int32> sizes)
   : MultiArray2<DataType>(sizes)
   {}
+  explicit UniqueMultiArray2(IMemoryAllocator* allocator)
+  : UniqueMultiArray2(MemoryAllocationOptions(allocator))
+  {}
+  explicit UniqueMultiArray2(const MemoryAllocationOptions& allocation_options)
+  : MultiArray2<DataType>(allocation_options)
+  {}
+  UniqueMultiArray2(const MemoryAllocationOptions& allocation_options,
+                    ConstArrayView<Int32> sizes)
+  : MultiArray2<DataType>(allocation_options, sizes)
+  {}
   UniqueMultiArray2(ConstMultiArray2View<DataType> view)
   : MultiArray2<DataType>(view)
   {}

arcane/src/arcane/utils/MultiArray2View.h

@@ -17,6 +17,8 @@
 #include "arcane/utils/UtilsTypes.h"
 #include "arcane/utils/ArrayView.h"
 
+#include <type_traits>
+
 /*---------------------------------------------------------------------------*/
 /*---------------------------------------------------------------------------*/
 
@@ -129,6 +131,60 @@ class ConstMultiArray2View
   ConstArrayView<Int32> m_sizes;
 };
 
+/*---------------------------------------------------------------------------*/
+/*---------------------------------------------------------------------------*/
+/*!
+ * \brief Vue sur un MultiArray2.
+ *
+ * Les instances de cette classe sont créées par appel à MultiArray2::span()
+ * ou MultiArray2::constSpan().
+ */
+template <class DataType>
+class MultiArray2SmallSpan
+{
+ private:
+
+  friend class MultiArray2<std::remove_cv_t<DataType>>;
+
+ public:
+
+  //! Vue vide
+  MultiArray2SmallSpan() = default;
+
+ private:
+
+  //! Vue sur la tableau \a buf
+  MultiArray2SmallSpan(SmallSpan<DataType> buf, SmallSpan<const Int32> indexes,
+                       SmallSpan<const Int32> sizes)
+  : m_buffer(buf)
+  , m_indexes(indexes)
+  , m_sizes(sizes)
+  {}
+
+ public:
+
+  //! Nombre d'éléments de la première dimension.
+  constexpr ARCCORE_HOST_DEVICE Int32 dim1Size() const { return m_sizes.size(); }
+  //! Nombre d'éléments de la deuxième dimension
+  constexpr ARCCORE_HOST_DEVICE SmallSpan<const Int32> dim2Sizes() const { return m_sizes; }
+  //! Nombre total d'éléments dans le tableau.
+  constexpr ARCCORE_HOST_DEVICE Int32 totalNbElement() const { return m_buffer.size(); }
+
+ public:
+
+  //! \a i-ème élément du tableau
+  constexpr ARCCORE_HOST_DEVICE SmallSpan<DataType> operator[](Int32 i) const
+  {
+    return m_buffer.subSpan(m_indexes[i], m_sizes[i]);
+  }
+
+ private:
+
+  SmallSpan<DataType> m_buffer;
+  SmallSpan<const Int32> m_indexes;
+  SmallSpan<const Int32> m_sizes;
+};
+
 /*---------------------------------------------------------------------------*/
 /*---------------------------------------------------------------------------*/
 

arcane/src/arcane/utils/tests/TestCollections.cc

@@ -11,6 +11,7 @@
 #include "arcane/utils/String.h"
 #include "arcane/utils/SmallArray.h"
 #include "arcane/utils/FixedArray.h"
+#include "arcane/utils/MultiArray2.h"
 
 #ifdef ARCANE_HAS_CXX20
 #include <ranges>
@@ -285,6 +286,10 @@ template class CollectionImplT<String>;
 template class SmallArray<Int32>;
 template class FixedArray<Int32,3>;
 template class FixedArray<double, 21>;
+
+template class MultiArray2<Int32>;
+template class UniqueMultiArray2<Int32>;
+template class SharedMultiArray2<Int32>;
 }
 
 /*---------------------------------------------------------------------------*/