File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change 31
31
# Increases the runtime closure size by ~700M
32
32
useMpi ? false ,
33
33
useRocm ? config . rocmSupport ,
34
+ rocmGpuTargets ? builtins . concatStringsSep ";" rocmPackages . clr . gpuTargets ,
34
35
enableCurl ? true ,
35
36
useVulkan ? false ,
36
37
llamaVersion ? "0.0.0" , # Arbitrary version, substituted by the flake
@@ -188,7 +189,7 @@ effectiveStdenv.mkDerivation (finalAttrs: {
188
189
]
189
190
++ optionals useRocm [
190
191
( cmakeFeature "CMAKE_HIP_COMPILER" "${ rocmPackages . llvm . clang } /bin/clang" )
191
- ( cmakeFeature "CMAKE_HIP_ARCHITECTURES" ( builtins . concatStringsSep ";" rocmPackages . clr . gpuTargets ) )
192
+ ( cmakeFeature "CMAKE_HIP_ARCHITECTURES" rocmGpuTargets )
192
193
]
193
194
++ optionals useMetalKit [
194
195
( lib . cmakeFeature "CMAKE_C_FLAGS" "-D__ARM_FEATURE_DOTPROD=1" )
You can’t perform that action at this time.
0 commit comments