pkgs/development/libraries/rocthrust/default.nix

   1 { lib
   2 , stdenv
   3 , fetchFromGitHub
   4 , cmake
   5 , rocm-cmake
   6 , rocm-runtime
   7 , rocm-device-libs
   8 , rocm-comgr
   9 , rocprim
  10 , hip
  11 , gtest ? null
  12 , buildTests ? false
  13 , buildBenchmarks ? false
  14 }:
  15
  16 assert buildTests -> gtest != null;
  17
  18 # Doesn't seem to work, thousands of errors compiling with no clear fix
  19 # Is this an upstream issue? We don't seem to be missing dependencies
  20 assert buildTests == false;
  21 assert buildBenchmarks == false;
  22
  23 stdenv.mkDerivation rec {
  24   pname = "rocthrust";
  25   rocmVersion = "5.3.1";
  26   version = "2.16.0-${rocmVersion}";
  27
  28   # Comment out these outputs until tests/benchmarks are fixed (upstream?)
  29   # outputs = [
  30   #   "out"
  31   # ] ++ lib.optionals buildTests [
  32   #   "test"
  33   # ] ++ lib.optionals buildBenchmarks [
  34   #   "benchmark"
  35   # ];
  36
  37   src = fetchFromGitHub {
  38     owner = "ROCmSoftwarePlatform";
  39     repo = "rocThrust";
  40     rev = "rocm-${rocmVersion}";
  41     hash = "sha256-cT0VyEVz86xR6qubAY2ncTxtCRTwXrNTWcFyf3mV+y0=";
  42   };
  43
  44   nativeBuildInputs = [
  45     cmake
  46     rocm-cmake
  47     rocprim
  48     hip
  49   ];
  50
  51   buildInputs = [
  52     rocm-runtime
  53     rocm-device-libs
  54     rocm-comgr
  55   ] ++ lib.optionals buildTests [
  56     gtest
  57   ];
  58
  59   cmakeFlags = [
  60     "-DCMAKE_CXX_COMPILER=hipcc"
  61     "-DHIP_ROOT_DIR=${hip}"
  62     # Manually define CMAKE_INSTALL_<DIR>
  63     # See: https://github.com/NixOS/nixpkgs/pull/197838
  64     "-DCMAKE_INSTALL_BINDIR=bin"
  65     "-DCMAKE_INSTALL_LIBDIR=lib"
  66     "-DCMAKE_INSTALL_INCLUDEDIR=include"
  67   ] ++ lib.optionals buildTests [
  68     "-DBUILD_TEST=ON"
  69   ] ++ lib.optionals buildBenchmarks [
  70     "-DBUILD_BENCHMARKS=ON"
  71   ];
  72
  73   # Comment out these outputs until tests/benchmarks are fixed (upstream?)
  74   # postInstall = lib.optionalString buildTests ''
  75   #   mkdir -p $test/bin
  76   #   mv $out/bin/test_* $test/bin
  77   # '' + lib.optionalString buildBenchmarks ''
  78   #   mkdir -p $benchmark/bin
  79   #   mv $out/bin/benchmark_* $benchmark/bin
  80   # '' + lib.optionalString (buildTests || buildBenchmarks) ''
  81   #   rmdir $out/bin
  82   # '';
  83
  84   meta = with lib; {
  85     description = "ROCm parallel algorithm library";
  86     homepage = "https://github.com/ROCmSoftwarePlatform/rocThrust";
  87     license = with licenses; [ asl20 ];
  88     maintainers = with maintainers; [ Madouura ];
  89     broken = rocmVersion != hip.version;
  90   };
  91 }