DGX added

This commit is contained in:
Caubet Serrabou Marc
2021-03-16 10:29:08 +01:00
parent 050398b556
commit 594b9deef5
2 changed files with 10 additions and 8 deletions
+7 -7
View File
@@ -28,11 +28,11 @@ pbuild::pre_configure() {
pbuild::add_configure_args "--with-cuda=$CUDA_HOME"
fi
if pbuild::use_flag slurm ; then
pbuild::add_configure_args "--without-gdrcopy" # Will be --with-gdrcopy when available
pbuild::add_configure_args "--with-knem"
pbuild::add_configure_args "--with-rdmacm"
else
pbuild::add_configure_args "--without-gdrcopy"
fi
# if pbuild::use_flag slurm || pbuild::use_flag dgx; then
pbuild::add_configure_args "--without-gdrcopy" # Will be --with-gdrcopy when available
pbuild::add_configure_args "--with-knem"
pbuild::add_configure_args "--with-rdmacm"
# else
# pbuild::add_configure_args "--without-gdrcopy"
# fi
}