diff --git a/pkgs/development/libraries/rocfft/default.nix b/pkgs/development/libraries/rocfft/default.nix
index 470097d2968..c89d08a82a1 100644
--- a/pkgs/development/libraries/rocfft/default.nix
+++ b/pkgs/development/libraries/rocfft/default.nix
@@ -2,6 +2,7 @@
, stdenv
, fetchFromGitHub
, rocmUpdateScript
+, runCommand
, cmake
, rocm-cmake
, hip
@@ -16,94 +17,139 @@
, buildBenchmarks ? false
}:
-stdenv.mkDerivation (finalAttrs: {
- pname = "rocfft";
- version = "5.4.2";
+let
+ rocfft = stdenv.mkDerivation (finalAttrs: {
+ pname = "rocfft";
+ version = "5.4.2";
- outputs = [
- "out"
- ] ++ lib.optionals buildTests [
- "test"
- ] ++ lib.optionals buildBenchmarks [
- "benchmark"
- ];
+ outputs = [
+ "out"
+ ] ++ lib.optionals buildTests [
+ "test"
+ ] ++ lib.optionals buildBenchmarks [
+ "benchmark"
+ ];
- src = fetchFromGitHub {
- owner = "ROCmSoftwarePlatform";
- repo = "rocFFT";
- rev = "rocm-${finalAttrs.version}";
- hash = "sha256-NsYeEoBQ/0z31ZQ32l7N+qavWEVkH37snkTHntot7nE=";
- };
+ src = fetchFromGitHub {
+ owner = "ROCmSoftwarePlatform";
+ repo = "rocFFT";
+ rev = "rocm-${finalAttrs.version}";
+ hash = "sha256-NsYeEoBQ/0z31ZQ32l7N+qavWEVkH37snkTHntot7nE=";
+ };
- nativeBuildInputs = [
- cmake
- rocm-cmake
- hip
- ];
+ nativeBuildInputs = [
+ cmake
+ rocm-cmake
+ hip
+ ];
- buildInputs = [
- sqlite
- python3
- ] ++ lib.optionals buildTests [
- gtest
- ] ++ lib.optionals (buildTests || buildBenchmarks) [
- boost
- fftw
- fftwFloat
- openmp
- ];
+ buildInputs = [
+ sqlite
+ python3
+ ] ++ lib.optionals buildTests [
+ gtest
+ ] ++ lib.optionals (buildTests || buildBenchmarks) [
+ boost
+ fftw
+ fftwFloat
+ openmp
+ ];
- propogatedBuildInputs = lib.optionals buildTests [
- fftw
- fftwFloat
- ];
+ propogatedBuildInputs = lib.optionals buildTests [
+ fftw
+ fftwFloat
+ ];
- cmakeFlags = [
- "-DCMAKE_C_COMPILER=hipcc"
- "-DCMAKE_CXX_COMPILER=hipcc"
- "-DUSE_HIP_CLANG=ON"
- "-DSQLITE_USE_SYSTEM_PACKAGE=ON"
- # 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_CLIENTS_TESTS=ON"
- ] ++ lib.optionals buildBenchmarks [
- "-DBUILD_CLIENTS_RIDER=ON"
- "-DBUILD_CLIENTS_SAMPLES=ON"
- ];
+ cmakeFlags = [
+ "-DCMAKE_C_COMPILER=hipcc"
+ "-DCMAKE_CXX_COMPILER=hipcc"
+ "-DUSE_HIP_CLANG=ON"
+ "-DSQLITE_USE_SYSTEM_PACKAGE=ON"
+ # 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_CLIENTS_TESTS=ON"
+ ] ++ lib.optionals buildBenchmarks [
+ "-DBUILD_CLIENTS_RIDER=ON"
+ "-DBUILD_CLIENTS_SAMPLES=ON"
+ ];
- postInstall = lib.optionalString buildTests ''
- mkdir -p $test/{bin,lib/fftw}
- cp -a $out/bin/* $test/bin
- ln -s ${fftw}/lib/libfftw*.so $test/lib/fftw
- ln -s ${fftwFloat}/lib/libfftw*.so $test/lib/fftw
- rm -r $out/lib/fftw
- rm $test/bin/{rocfft_rtc_helper,*-rider} || true
- '' + lib.optionalString buildBenchmarks ''
- mkdir -p $benchmark/bin
- cp -a $out/bin/* $benchmark/bin
- rm $benchmark/bin/{rocfft_rtc_helper,*-test} || true
- '' + lib.optionalString (buildTests || buildBenchmarks ) ''
- mv $out/bin/rocfft_rtc_helper $out
- rm -r $out/bin/*
- mv $out/rocfft_rtc_helper $out/bin
+ postInstall = lib.optionalString buildTests ''
+ mkdir -p $test/{bin,lib/fftw}
+ cp -a $out/bin/* $test/bin
+ ln -s ${fftw}/lib/libfftw*.so $test/lib/fftw
+ ln -s ${fftwFloat}/lib/libfftw*.so $test/lib/fftw
+ rm -r $out/lib/fftw
+ rm $test/bin/{rocfft_rtc_helper,*-rider} || true
+ '' + lib.optionalString buildBenchmarks ''
+ mkdir -p $benchmark/bin
+ cp -a $out/bin/* $benchmark/bin
+ rm $benchmark/bin/{rocfft_rtc_helper,*-test} || true
+ '' + lib.optionalString (buildTests || buildBenchmarks ) ''
+ mv $out/bin/rocfft_rtc_helper $out
+ rm -r $out/bin/*
+ mv $out/rocfft_rtc_helper $out/bin
+ '';
+
+ passthru.updateScript = rocmUpdateScript {
+ name = finalAttrs.pname;
+ owner = finalAttrs.src.owner;
+ repo = finalAttrs.src.repo;
+ };
+
+ meta = with lib; {
+ description = "FFT implementation for ROCm ";
+ homepage = "https://github.com/ROCmSoftwarePlatform/rocFFT";
+ license = with licenses; [ mit ];
+ maintainers = teams.rocm.members;
+ broken = versions.minor finalAttrs.version != versions.minor hip.version;
+ };
+ });
+
+ rocfft-zero = runCommand "rocfft-zero" { preferLocalBuild = true; } ''
+ mkdir -p $out
+ cp -a ${rocfft}/lib/librocfft-device-0* $out
'';
- passthru.updateScript = rocmUpdateScript {
- name = finalAttrs.pname;
- owner = finalAttrs.src.owner;
- repo = finalAttrs.src.repo;
- };
+ rocfft-one = runCommand "rocfft-one" { preferLocalBuild = true; } ''
+ mkdir -p $out
+ cp -a ${rocfft}/lib/librocfft-device-1* $out
+ '';
- meta = with lib; {
- description = "FFT implementation for ROCm ";
- homepage = "https://github.com/ROCmSoftwarePlatform/rocFFT";
- license = with licenses; [ mit ];
- maintainers = teams.rocm.members;
- broken = versions.minor finalAttrs.version != versions.minor hip.version;
- hydraPlatforms = [ ]; # rocFFT produces an extremely large output
- };
-})
+ rocfft-two = runCommand "rocfft-two" { preferLocalBuild = true; } ''
+ mkdir -p $out
+ cp -a ${rocfft}/lib/librocfft-device-2* $out
+ '';
+
+ rocfft-three = runCommand "rocfft-three" { preferLocalBuild = true; } ''
+ mkdir -p $out
+ cp -a ${rocfft}/lib/librocfft-device-3* $out
+ '';
+in stdenv.mkDerivation {
+ inherit (rocfft) pname version outputs src passthru meta;
+
+ dontUnpack = true;
+ dontConfigure = true;
+ dontBuild = true;
+
+ installPhase = ''
+ runHook preInstall
+
+ mkdir -p $out/lib
+
+ for path in ${rocfft-zero} ${rocfft-one} ${rocfft-two} ${rocfft-three}; do
+ cp -as $path/* $out/lib
+ done
+
+ cp -an ${rocfft}/* $out
+ '' + lib.optionalString buildTests ''
+ cp -a ${rocfft.test} $test
+ '' + lib.optionalString buildBenchmarks ''
+ cp -a ${rocfft.benchmark} $benchmark
+ '' + ''
+ runHook postInstall
+ '';
+}