diff --git a/pkgs/development/libraries/rocprim/default.nix b/pkgs/development/libraries/rocprim/default.nix new file mode 100644 index 000000000000..97b8daf8a7b1 --- /dev/null +++ b/pkgs/development/libraries/rocprim/default.nix @@ -0,0 +1,85 @@ +{ lib +, stdenv +, fetchFromGitHub +, cmake +, rocm-cmake +, rocm-runtime +, rocm-device-libs +, rocm-comgr +, hip +, gtest ? null +, gbenchmark ? null +, buildTests ? false +, buildBenchmarks ? false +}: + +assert buildTests -> gtest != null; +assert buildBenchmarks -> gbenchmark != null; + +stdenv.mkDerivation rec { + pname = "rocprim"; + rocmVersion = "5.3.1"; + version = "2.11.0-${rocmVersion}"; + + outputs = [ + "out" + ] ++ lib.optionals buildTests [ + "test" + ] ++ lib.optionals buildBenchmarks [ + "benchmark" + ]; + + src = fetchFromGitHub { + owner = "ROCmSoftwarePlatform"; + repo = "rocPRIM"; + rev = "rocm-${rocmVersion}"; + hash = "sha256-aapvj9bwwlg7VJfnH1PVR8DulMcJh1xR6B4rPPGU6Q4="; + }; + + nativeBuildInputs = [ + cmake + rocm-cmake + hip + ]; + + buildInputs = [ + rocm-runtime + rocm-device-libs + rocm-comgr + ] ++ lib.optionals buildTests [ + gtest + ] ++ lib.optionals buildBenchmarks [ + gbenchmark + ]; + + cmakeFlags = [ + "-DCMAKE_CXX_COMPILER=hipcc" + # Manually define CMAKE_INSTALL_ + # See: https://github.com/NixOS/nixpkgs/pull/197838 + "-DCMAKE_INSTALL_BINDIR=bin" + "-DCMAKE_INSTALL_LIBDIR=lib" + "-DCMAKE_INSTALL_INCLUDEDIR=include" + ] ++ lib.optionals buildTests [ + "-DBUILD_TEST=ON" + ] ++ lib.optionals buildBenchmarks [ + "-DBUILD_BENCHMARK=ON" + ]; + + postInstall = lib.optionalString buildTests '' + mkdir -p $test/bin + mv $out/bin/test_* $test/bin + '' + lib.optionalString buildBenchmarks '' + mkdir -p $benchmark/bin + mv $out/bin/benchmark_* $benchmark/bin + '' + lib.optionalString (buildTests || buildBenchmarks) '' + rmdir $out/bin + ''; + + meta = with lib; { + description = "ROCm parallel primitives"; + homepage = "https://github.com/ROCmSoftwarePlatform/rocPRIM"; + license = with licenses; [ mit ]; + maintainers = with maintainers; [ Madouura ]; + broken = rocmVersion != hip.version; + }; +} diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 0b826aad5436..c772e58389f3 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -14877,6 +14877,8 @@ with pkgs; rocmlir = callPackage ../development/libraries/rocmlir { }; + rocprim = callPackage ../development/libraries/rocprim { }; + rtags = callPackage ../development/tools/rtags { inherit (darwin) apple_sdk; };