rarias/devshell
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Configure MPICH for MN5

Browse Source
This commit is contained in:
Rodrigo Arias 2025-10-30 10:44:27 +01:00
parent 676a0ced1c
commit 9bae257774
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
isabel/slurm/flake.nix
View File

@ -3,6 +3,7 @@
outputs = { self, jungle }: outputs = { self, jungle }:
let let
nixpkgs = jungle.inputs.nixpkgs; nixpkgs = jungle.inputs.nixpkgs;
lib = nixpkgs.lib;
clusterOverlay = (final: prev: { clusterOverlay = (final: prev: {
# Use cluster llvm compiler repo # Use cluster llvm compiler repo
clangOmpss2Unwrapped = prev.clangOmpss2Unwrapped.override { clangOmpss2Unwrapped = prev.clangOmpss2Unwrapped.override {
@ -11,6 +12,13 @@
gitCommit = "151c260ba834826c01855da0a41fc203ffe4d025"; gitCommit = "151c260ba834826c01855da0a41fc203ffe4d025";
gitBranch = "cluster"; gitBranch = "cluster";
}; };
# Configure MPICH to use UCX with multiple thread support
mpich = prev.mpich.overrideAttrs (old: {
configureFlags = (lib.remove "--with-device=ch4:ofi" old.configureFlags) ++ [
"--with-device=ch4:ucx"
"--enable-threads=multiple"
];
});
}); });
pkgs = import nixpkgs { pkgs = import nixpkgs {
system = "x86_64-linux"; system = "x86_64-linux";
@ -26,7 +34,7 @@
pname = "devshell"; pname = "devshell";
buildInputs = with pkgs; [ buildInputs = with pkgs; [
slurm.out slurm.dev gcc slurm.out slurm.dev gcc
clangOmpss2 clangOmpss2 mpich osumb
]; ];
inputsFrom = with pkgs; [ inputsFrom = with pkgs; [
nanos6 nanos6