chn/nixos
1
0
Fork 0
mirror of https://github.com/CHN-beta/nixos.git synced 2026-01-12 01:29:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

move nvidia mpi source

Browse Source
This commit is contained in:
陈浩南
2025-04-12 13:46:15 +08:00
parent f5bf96e4b0
commit ef0829f76a
3 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
flake/src.nix
View File

@@ -8,6 +8,14 @@
url = "https://developer.download.nvidia.com/hpc-sdk/24.11/nvhpc_2024_2411_Linux_x86_64_cuda_12.6.tar.gz";
sha256 = "080rb89p2z98b75wqssvp3s8x6b5n0556d0zskh3cfapcb08lh1r";
};
mpi = pkgs.requireFile
{
name = "openmpi-gitclone.tar.gz";
# download from https://developer.nvidia.com/networking/hpc-x/eula?mrequest=downloads&mtype=hpc&mver=hpc-x&mname=v2.22/hpcx-v2.22-gcc-doca_ofed-ubuntu24.04-cuda12-x86_64.tbz
# nix-prefetch-url file://$(pwd)/openmpi-gitclone.tar.gz
sha256 = "05r5x6mgw2f2kcq9vhdkfj42panchzlbpns8qy57y4jsbmabwabi";
message = "Source file not found.";
};
version = "24.11";
cudaVersion = "12.6";
};

3
packages/default.nix
View File

@@ -107,7 +107,8 @@ inputs: rec
fmt = (inputs.pkgs.fmt.override { inherit (final) stdenv; }).overrideAttrs { doCheck = false; };
hdf5 = inputs.pkgs.hdf5.override
{ inherit (final) stdenv; cppSupport = false; fortranSupport = true; enableShared = false; enableStatic = true; };
mpi = inputs.pkgs.callPackage ./nvhpc/mpi.nix { inherit (final) stdenv; };
mpi = inputs.pkgs.callPackage ./nvhpc/mpi.nix
{ inherit (final) stdenv; src = inputs.topInputs.self.src.nvhpc.mpi; };
});
gccFull = inputs.pkgs.symlinkJoin
{

11
packages/nvhpc/mpi.nix
View File

@@ -1,18 +1,11 @@
# pick from nixpkgs 24e16d8b21f698cbe372be67b645a1919bfd0d20
{
requireFile, stdenv, lib,
requireFile, stdenv, lib, src,
perl, libnl, rdma-core, zlib, numactl, libevent, hwloc, libpsm2, libfabric, pmix, ucx, ucc, prrte
}: stdenv.mkDerivation rec
{
name = "openmpi";
src = requireFile
{
name = "openmpi-gitclone.tar.gz";
# download from https://developer.nvidia.com/networking/hpc-x/eula?mrequest=downloads&mtype=hpc&mver=hpc-x&mname=v2.22/hpcx-v2.22-gcc-doca_ofed-ubuntu24.04-cuda12-x86_64.tbz
# nix-prefetch-url file://$(pwd)/openmpi-gitclone.tar.gz
sha256 = "05r5x6mgw2f2kcq9vhdkfj42panchzlbpns8qy57y4jsbmabwabi";
message = "Source file not found.";
};
inherit src;
postPatch = ''
patchShebangs ./
${lib.pipe {