Skip to content

Commit ae6bebc

Browse files
committed
nix: respect config.cudaCapabilities
1 parent 4522c47 commit ae6bebc

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

.devops/nix/package.nix

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,14 @@ effectiveStdenv.mkDerivation (
150150
(cmakeBool "LLAMA_HIPBLAS" useRocm)
151151
(cmakeBool "LLAMA_METAL" useMetalKit)
152152
]
153+
++ optionals useCuda [
154+
(
155+
with cudaPackages.flags;
156+
cmakeFeature "CMAKE_CUDA_ARCHITECTURES" (
157+
builtins.concatStringsSep ";" (map dropDot cudaCapabilities)
158+
)
159+
)
160+
]
153161
++ optionals useRocm [
154162
(cmakeFeature "CMAKE_C_COMPILER" "hipcc")
155163
(cmakeFeature "CMAKE_CXX_COMPILER" "hipcc")

0 commit comments

Comments
 (0)