Skip to content

Commit

Permalink
koboldcpp: fix CUDA_DOCKER_ARCH make flag
Browse files Browse the repository at this point in the history
- Previously, the value was "assembled" using additional logic, but
  when switching to a different method, an oversight was made, and
  `sm_` was forgotten to be removed
  • Loading branch information
DontEatOreo committed Jul 23, 2024
1 parent f9e9e91 commit db9fe54
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions pkgs/by-name/ko/koboldcpp/package.nix
Original file line number Diff line number Diff line change
Expand Up @@ -129,8 +129,7 @@ effectiveStdenv.mkDerivation (finalAttrs: {
(makeBool "LLAMA_CLBLAST" clblastSupport)
(makeBool "LLAMA_VULKAN" vulkanSupport)
(makeBool "LLAMA_METAL" metalSupport)
(lib.optionals cublasSupport "CUDA_DOCKER_ARCH=sm_${builtins.head cudaArches}")
];
] ++ lib.optionals cublasSupport [ "CUDA_DOCKER_ARCH=${builtins.head cudaArches}" ];

installPhase = ''
runHook preInstall
Expand Down

0 comments on commit db9fe54

Please sign in to comment.