From 26e2f2d109945ac8e4027b07a26813d080638b9e Mon Sep 17 00:00:00 2001 From: Alejandro Gallo Date: Mon, 5 Dec 2022 17:49:54 +0100 Subject: [PATCH] Add ATRIP_SOURCES_IN_GPU and ATRIP_CUDA_AWARE_MPI defines in configure --- configure.ac | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/configure.ac b/configure.ac index 0679d32..4bd956b 100644 --- a/configure.ac +++ b/configure.ac @@ -90,6 +90,17 @@ ATRIP_DEF([naive-slow], [enable], [Run slow but correct code for the mapping of (iteration, rank) to tuple of the naive tuple distribution.]) +ATRIP_DEF([sources-in-gpu], [enable], + [ATRIP_SOURCES_IN_GPU], + [When using CUDA, activate storing all sources (slices of + the input tensors) in the GPU. This means that a lot of GPUs + will be needed.]) + +ATRIP_DEF([cuda-aware-mpi], [enable], + [ATRIP_CUDA_AWARE_MPI], + [When using MPI, assume support for CUDA aware mpi by the + given MPI implementation.]) + dnl -----------------------------------------------------------------------