We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4522c47 commit ae6bebcCopy full SHA for ae6bebc
.devops/nix/package.nix
@@ -150,6 +150,14 @@ effectiveStdenv.mkDerivation (
150
(cmakeBool "LLAMA_HIPBLAS" useRocm)
151
(cmakeBool "LLAMA_METAL" useMetalKit)
152
]
153
+ ++ optionals useCuda [
154
+ (
155
+ with cudaPackages.flags;
156
+ cmakeFeature "CMAKE_CUDA_ARCHITECTURES" (
157
+ builtins.concatStringsSep ";" (map dropDot cudaCapabilities)
158
+ )
159
160
+ ]
161
++ optionals useRocm [
162
(cmakeFeature "CMAKE_C_COMPILER" "hipcc")
163
(cmakeFeature "CMAKE_CXX_COMPILER" "hipcc")
0 commit comments