Unverified Commit 3c405967 authored by Philip Taron's avatar Philip Taron Committed by GitHub
Browse files

llama-cpp: allow to override ROCM GPU targets (#435263)

parents a7e204e5 f9696ce3

pkgs/by-name/ll/llama-cpp/package.nix

+2 −2
Original line number Diff line number Diff line
@@ -12,6 +12,7 @@ 


  rocmSupport ? config.rocmSupport,

  rocmPackages ? { },

  rocmGpuTargets ? builtins.concatStringsSep ";" rocmPackages.clr.gpuTargets,



  openclSupport ? false,

  clblast,
@@ -146,8 +147,7 @@ effectiveStdenv.mkDerivation (finalAttrs: { 
  ]

  ++ optionals rocmSupport [

    (cmakeFeature "CMAKE_HIP_COMPILER" "${rocmPackages.clr.hipClangPath}/clang++")

    # TODO: this should become `clr.gpuTargets` in the future.

    (cmakeFeature "CMAKE_HIP_ARCHITECTURES" rocmPackages.rocblas.amdgpu_targets)

    (cmakeFeature "CMAKE_HIP_ARCHITECTURES" rocmGpuTargets)

  ]

  ++ optionals metalSupport [

    (cmakeFeature "CMAKE_C_FLAGS" "-D__ARM_FEATURE_DOTPROD=1")