Unverified Commit db45f307 authored by Tristan Ross's avatar Tristan Ross
Browse files

spirv-llvm-translator: fix formatting

parent 8bad5091

pkgs/development/compilers/spirv-llvm-translator/default.nix

+121 −89
Original line number Diff line number Diff line 
{ lib, stdenv

, fetchFromGitHub

, fetchpatch

, cmake

, pkg-config

, lit

, llvm

, spirv-headers

, spirv-tools

{

  lib,

  stdenv,

  fetchFromGitHub,

  fetchpatch,

  cmake,

  pkg-config,

  lit,

  llvm,

  spirv-headers,

  spirv-tools,

}:



let
@@ -15,35 +17,50 @@ let 


  # ROCm, if actively updated will always be at the latest version

  branch =

    if llvmMajor == "19" then rec {

    if llvmMajor == "19" then

      rec {

        version = "19.1.0";

        rev = "dad1f0eaab8047a4f73c50ed5f3d1694b78aae97";

        hash = "sha256-mUvDF5y+cBnqUaHjyiiE8cJGH5MfQMqGFy6bYv9vCVY=";

    } else if llvmMajor == "18" then rec {

      }

    else if llvmMajor == "18" then

      rec {

        version = "18.1.0";

        rev = "v${version}";

        hash = "sha256-64guZiuO7VpaX01wNIjV7cnjEAe6ineMdY44S6sA33k=";

    } else if llvmMajor == "17" || isROCm then rec {

      }

    else if llvmMajor == "17" || isROCm then

      rec {

        version = "17.0.0";

        rev = "v${version}";

        hash = "sha256-Rzm5Py9IPFtS9G7kME+uSwZ/0gPGW6MlL35ZWk4LfHM=";

    } else if llvmMajor == "16" then rec {

      }

    else if llvmMajor == "16" then

      rec {

        version = "16.0.0";

        rev = "v${version}";

        hash = "sha256-EUabcYqSjXshbPmcs1DRLvCSL1nd9rEdpqELBrItCW8=";

    } else if llvmMajor == "15" then rec {

      }

    else if llvmMajor == "15" then

      rec {

        version = "15.0.0";

        rev = "v${version}";

        hash = "sha256-OsDohXRxovtEXaWiRGp8gJ0dXmoALyO+ZimeSO8aPVI=";

    } else if llvmMajor == "14" then {

      }

    else if llvmMajor == "14" then

      {

        version = "14.0.0+unstable-2024-07-15";

        rev = "2823e7052b7999c10fff63bc8089e5aa205716f4";

        hash = "sha256-8/4B74hYge6WiH7PzRGEgE3W7f9IkQ4VMmfkWKYA/l4=";

    } else if llvmMajor == "11" then {

      }

    else if llvmMajor == "11" then

      {

        version = "11.0.0+unstable-2022-05-04";

        rev = "4ef524240833abfeee1c5b9fff6b1bd53f4806b3"; # 267 commits ahead of v11.0.0

        hash = "sha256-NoIoa20+2sH41rEnr8lsMhtfesrtdPINiXtUnxYVm8s=";

    } else throw "Incompatible LLVM version.";

      }

    else

      throw "Incompatible LLVM version.";

in

stdenv.mkDerivation {

  pname = "SPIRV-LLVM-Translator";
@@ -55,7 +72,8 @@ stdenv.mkDerivation { 
    inherit (branch) rev hash;

  };



  patches = lib.optionals (llvmMajor == "18") [

  patches =

    lib.optionals (llvmMajor == "18") [

      # Fixes build after SPV_INTEL_maximum_registers breaking change

      # TODO: remove on next spirv-headers release

      (fetchpatch {
@@ -63,20 +81,23 @@ stdenv.mkDerivation { 
        revert = true;

        hash = "sha256-71sJuGqVjTcB549eIiCO0LoqAgxkdEHCoxh8Pd/Qzz8=";

      })

  ] ++ lib.optionals (lib.versionAtLeast llvmMajor "15" && lib.versionOlder llvmMajor "18") [

    ]

    ++ lib.optionals (lib.versionAtLeast llvmMajor "15" && lib.versionOlder llvmMajor "18") [

      # Fixes build after spirv-headers breaking change

      (fetchpatch {

        url = "https://github.com/KhronosGroup/SPIRV-LLVM-Translator/commit/0166a0fb86dc6c0e8903436bbc3a89bc3273ebc0.patch";

        excludes = [ "spirv-headers-tag.conf" ];

        hash = "sha256-17JJG8eCFVphElY5fVT/79hj0bByWxo8mVp1ZNjQk/M=";

      })

  ] ++ lib.optionals (llvmMajor == "16") [

    ]

    ++ lib.optionals (llvmMajor == "16") [

      # Fixes builds that link against external LLVM dynamic library

      (fetchpatch {

        url = "https://github.com/KhronosGroup/SPIRV-LLVM-Translator/commit/f3b9b604d7eda18d0d1029d94a6eebd33aa3a3fe.patch";

        hash = "sha256-opDjyZcy7O4wcSfm/A51NCIiDyIvbcmbv9ns1njdJbc=";

      })

  ] ++ lib.optionals (llvmMajor == "14") [

    ]

    ++ lib.optionals (llvmMajor == "14") [

      (fetchpatch {

        # tries to install llvm-spirv into llvm nix store path

        url = "https://github.com/KhronosGroup/SPIRV-LLVM-Translator/commit/cce9a2f130070d799000cac42fe24789d2b777ab.patch";
@@ -85,32 +106,43 @@ stdenv.mkDerivation { 
      })

    ];



  nativeBuildInputs = [ pkg-config cmake ]

    ++ (if isROCm then [ llvm ] else [ llvm.dev ]);

  nativeBuildInputs = [

    pkg-config

    cmake

  ] ++ (if isROCm then [ llvm ] else [ llvm.dev ]);



  buildInputs = [ spirv-headers spirv-tools ]

    ++ lib.optionals (!isROCm) [ llvm ];

  buildInputs = [

    spirv-headers

    spirv-tools

  ] ++ lib.optionals (!isROCm) [ llvm ];



  nativeCheckInputs = [ lit ];



  cmakeFlags = [

  cmakeFlags =

    [

      "-DLLVM_INCLUDE_TESTS=ON"

      "-DLLVM_DIR=${(if isROCm then llvm else llvm.dev)}"

      "-DBUILD_SHARED_LIBS=YES"

      "-DLLVM_SPIRV_BUILD_EXTERNAL=YES"

      # RPATH of binary /nix/store/.../bin/llvm-spirv contains a forbidden reference to /build/

      "-DCMAKE_SKIP_BUILD_RPATH=ON"

  ] ++ lib.optional (llvmMajor != "11") "-DLLVM_EXTERNAL_SPIRV_HEADERS_SOURCE_DIR=${spirv-headers.src}"

    ]

    ++ lib.optional (llvmMajor != "11") "-DLLVM_EXTERNAL_SPIRV_HEADERS_SOURCE_DIR=${spirv-headers.src}"

    ++ lib.optional (llvmMajor == "19") "-DBASE_LLVM_VERSION=${lib.versions.majorMinor llvm.version}.0";



  # FIXME: CMake tries to run "/llvm-lit" which of course doesn't exist

  doCheck = false;



  makeFlags = [ "all" "llvm-spirv" ];

  makeFlags = [

    "all"

    "llvm-spirv"

  ];



  postInstall = ''

  postInstall =

    ''

      install -D tools/llvm-spirv/llvm-spirv $out/bin/llvm-spirv

  '' + lib.optionalString stdenv.hostPlatform.isDarwin ''

    ''

    + lib.optionalString stdenv.hostPlatform.isDarwin ''

      install_name_tool $out/bin/llvm-spirv \

        -change @rpath/libLLVMSPIRVLib.dylib $out/lib/libLLVMSPIRVLib.dylib

    '';