[arcane,main] Ajoute possibilité de changer la ressource mémoire des données via une variable d'environnement.

Author: grospelliergilles

arcane/src/arcane/accelerator/core/AcceleratorRuntimeInitialisationInfo.cc

@@ -16,6 +16,7 @@
 #include "arcane/utils/ITraceMng.h"
 #include "arcane/utils/String.h"
 #include "arcane/utils/Property.h"
+#include "arcane/utils/MemoryUtils.h"
 
 #include "arcane/accelerator/core/Runner.h"
 #include "arcane/accelerator/core/DeviceId.h"
@@ -175,6 +176,7 @@ arcaneInitializeRunner(Accelerator::Runner& runner,ITraceMng* tm,
   if (policy==eExecutionPolicy::None)
     ARCANE_FATAL("Invalid policy eExecutionPolicy::None");
   tm->info() << "AcceleratorRuntime=" << accelerator_runtime;
+  tm->info() << "DefaultDataAllocator MemoryResource=" << MemoryUtils::getDefaultDataMemoryResource();
   if (impl::isAcceleratorPolicy(policy)){
     tm->info() << "Using accelerator runtime=" << policy << " device=" << acc_info.deviceId();
     runner.initialize(policy,acc_info.deviceId());

arcane/src/arcane/impl/ArcaneMain.cc

@@ -35,6 +35,7 @@
 #include "arcane/utils/CommandLineArguments.h"
 #include "arcane/utils/ApplicationInfo.h"
 #include "arcane/utils/TestLogger.h"
+#include "arcane/utils/MemoryUtils.h"
 #include "arcane/utils/internal/MemoryUtilsInternal.h"
 
 #include "arcane/core/ArcaneException.h"
@@ -1172,6 +1173,14 @@ _checkAutoDetectAccelerator(bool& has_accelerator)
 
     (*my_functor)(runtime_info);
     has_accelerator = true;
+
+    // Permet de surcharger le choix de l'allocateur des données
+    String data_allocator_str = Arcane::platform::getEnvironmentVariable("ARCANE_DEFAULT_DATA_MEMORY_RESOURCE");
+    if (!data_allocator_str.null()){
+      eMemoryResource v = MemoryUtils::getMemoryResourceFromName(data_allocator_str);
+      if (v!=eMemoryResource::Unknown)
+        MemoryUtils::setDefaultDataMemoryResource(v);
+    }
   }
   catch (const Exception& ex) {
     return arcanePrintArcaneException(ex, nullptr);