File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 10
10
git ,
11
11
python3 ,
12
12
mpi ,
13
- openblas , # TODO: Use the generic ` blas` so users could switch between alternative implementations
13
+ blas ,
14
14
cudaPackages ,
15
15
darwin ,
16
16
rocmPackages ,
@@ -181,6 +181,7 @@ effectiveStdenv.mkDerivation (
181
181
++ optionals useMpi [ mpi ]
182
182
++ optionals useOpenCL [ clblast ]
183
183
++ optionals useRocm rocmBuildInputs
184
+ ++ optionals useBlas [ blas ]
184
185
++ optionals useVulkan vulkanBuildInputs ;
185
186
186
187
cmakeFlags =
@@ -216,8 +217,7 @@ effectiveStdenv.mkDerivation (
216
217
# Should likely use `rocmPackages.clr.gpuTargets`.
217
218
"-DAMDGPU_TARGETS=gfx803;gfx900;gfx906:xnack-;gfx908:xnack-;gfx90a:xnack+;gfx90a:xnack-;gfx940;gfx941;gfx942;gfx1010;gfx1012;gfx1030;gfx1100;gfx1101;gfx1102"
218
219
]
219
- ++ optionals useMetalKit [ ( lib . cmakeFeature "CMAKE_C_FLAGS" "-D__ARM_FEATURE_DOTPROD=1" ) ]
220
- ++ optionals useBlas [ ( lib . cmakeFeature "LLAMA_BLAS_VENDOR" "OpenBLAS" ) ] ;
220
+ ++ optionals useMetalKit [ ( lib . cmakeFeature "CMAKE_C_FLAGS" "-D__ARM_FEATURE_DOTPROD=1" ) ] ;
221
221
222
222
# TODO(SomeoneSerge): It's better to add proper install targets at the CMake level,
223
223
# if they haven't been added yet.
You can’t perform that action at this time.
0 commit comments