pkgs/by-name/um/umpire/package.nix

   1 { stdenv
   2 , lib
   3 , fetchFromGitHub
   4 , cmake
   5 , config
   6 , cudaSupport ? config.cudaSupport
   7 , cudaPackages ? null
   8 }:
   9
  10 assert cudaSupport -> cudaPackages != null;
  11
  12 stdenv.mkDerivation rec {
  13   pname = "umpire";
  14   version = "2024.07.0";
  15
  16   src = fetchFromGitHub {
  17     owner = "LLNL";
  18     repo = "umpire";
  19     rev = "v${version}";
  20     hash = "sha256-JbYaJe4bqlB272aZxB3Amw8fX/pmZr/4/7kaukAiK8c=";
  21     fetchSubmodules = true;
  22   };
  23
  24   nativeBuildInputs = [
  25     cmake
  26   ] ++ lib.optionals cudaSupport [
  27     cudaPackages.cuda_nvcc
  28   ];
  29
  30   buildInputs = lib.optionals cudaSupport (with cudaPackages; [
  31     cudatoolkit
  32     cuda_cudart
  33   ]);
  34
  35   cmakeFlags = lib.optionals cudaSupport [
  36     "-DCUDA_TOOLKIT_ROOT_DIR=${cudaPackages.cudatoolkit}"
  37     "-DENABLE_CUDA=ON"
  38     (lib.cmakeFeature "CMAKE_CUDA_ARCHITECTURES" cudaPackages.flags.cmakeCudaArchitecturesString)
  39   ];
  40
  41   meta = with lib; {
  42     description = "Application-focused API for memory management on NUMA & GPU architectures";
  43     homepage = "https://github.com/LLNL/Umpire";
  44     maintainers = with maintainers; [ sheepforce ];
  45     license = with licenses; [ mit ];
  46     platforms = [ "x86_64-linux" ];
  47   };
  48 }