1
0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-09 23:45:36 +00:00

Updated config options for MPI not being aware of GPU

This commit is contained in:
Peter Boyle 2024-03-26 00:42:00 +00:00
parent 5404fc66ab
commit f32c275376

View File

@ -226,23 +226,14 @@ case ${ac_SFW_FP16} in
esac esac
############### Default to accelerator cshift, but revert to host if UCX is buggy or other reasons ############### Default to accelerator cshift, but revert to host if UCX is buggy or other reasons
AC_ARG_ENABLE([accelerator-cshift], AC_ARG_ENABLE([accelerator-aware-mpi],
[AS_HELP_STRING([--enable-accelerator-cshift=yes|no],[run cshift on the device])], [AS_HELP_STRING([--enable-accelerator-aware-mpi=yes|no],[run mpi transfers from device])],
[ac_ACC_CSHIFT=${enable_accelerator_cshift}], [ac_ACC_CSHIFT=yes]) [ac_ACCELERATOR_AWARE_MPI=${enable_accelerator_aware_mpi}], [ac_ACCELERATOR_AWARE_MPI=yes])
AC_ARG_ENABLE([ucx-buggy], case ${ac_ACCELERATOR_AWARE_MPI} in
[AS_HELP_STRING([--enable-ucx-buggy=yes|no],[enable workaround for UCX device buffer bugs])],
[ac_UCXBUGGY=${enable_ucx_buggy}], [ac_UCXBUGGY=no])
case ${ac_UCXBUGGY} in
yes) yes)
ac_ACC_CSHIFT=no;; AC_DEFINE([ACCELERATOR_CSHIFT],[1],[ Cshift runs on host])
*);; AC_DEFINE([ACCELERATOR_AWARE_MPI],[1],[ Stencil can use device pointers]);;
esac
case ${ac_ACC_CSHIFT} in
yes)
AC_DEFINE([ACCELERATOR_CSHIFT],[1],[ UCX device buffer bugs are not present]);;
*);; *);;
esac esac