mirror of
https://github.com/CHN-beta/nixos.git
synced 2024-10-23 07:08:45 +08:00
23 lines
992 B
Nix
23 lines
992 B
Nix
inputs:
|
|
{
|
|
options.nixos.packages.lammps = let inherit (inputs.lib) mkOption types; in mkOption
|
|
{
|
|
type = types.nullOr (types.submodule {});
|
|
default = if inputs.config.nixos.system.gui.enable then {} else null;
|
|
};
|
|
config = let inherit (inputs.config.nixos.packages) lammps; in inputs.lib.mkIf (lammps != null)
|
|
{
|
|
nixos.packages.packages._packages =
|
|
let cuda = let inherit (inputs.config.nixos.system.nixpkgs) cuda; in cuda.enable && cuda.capabilities != null;
|
|
in
|
|
if cuda then [((inputs.pkgs.lammps.override { stdenv = inputs.pkgs.cudaPackages.backendStdenv; })
|
|
.overrideAttrs (prev:
|
|
{
|
|
cmakeFlags = prev.cmakeFlags ++ [ "-DPKG_GPU=on" "-DGPU_API=cuda" "-DCMAKE_POLICY_DEFAULT_CMP0146=OLD" ];
|
|
nativeBuildInputs = prev.nativeBuildInputs ++ [ inputs.pkgs.cudaPackages.cudatoolkit ];
|
|
buildInputs = prev.buildInputs ++ [ inputs.pkgs.mpi ];
|
|
}))]
|
|
else [ inputs.pkgs.lammps-mpi ];
|
|
};
|
|
}
|