File tree 2 files changed +12
-10
lines changed
2 files changed +12
-10
lines changed Original file line number Diff line number Diff line change 11
11
meta . mainProgram = "llama" ;
12
12
inherit ( pkgs . stdenv ) isAarch32 isAarch64 isDarwin ;
13
13
buildInputs = with pkgs ; [ openmpi ] ;
14
- osSpecific = with pkgs ; buildInputs ++
15
- (
14
+ osSpecific = with pkgs ; buildInputs ++ (
16
15
if isAarch64 && isDarwin then
17
16
with pkgs . darwin . apple_sdk_11_0 . frameworks ; [
18
17
Accelerate
96
95
} ;
97
96
packages . rocm = pkgs . stdenv . mkDerivation {
98
97
inherit name src meta postPatch nativeBuildInputs postInstall ;
99
- buildInputs = with pkgs ; buildInputs ++ [ hip hipblas rocblas ] ;
98
+ buildInputs = with pkgs . rocmPackages ; buildInputs ++ [ clr hipblas rocblas ] ;
100
99
cmakeFlags = cmakeFlags ++ [
101
100
"-DLLAMA_HIPBLAS=1"
102
101
"-DCMAKE_C_COMPILER=hipcc"
103
102
"-DCMAKE_CXX_COMPILER=hipcc"
104
- "-DCMAKE_POSITION_INDEPENDENT_CODE=ON"
103
+ # Build all targets supported by rocBLAS. When updating search for TARGET_LIST_ROCM
104
+ # in github.com/ROCmSoftwarePlatform/rocBLAS/blob/develop/CMakeLists.txt
105
+ # and select the line that matches the current nixpkgs version of rocBLAS.
106
+ "-DAMDGPU_TARGETS=gfx803;gfx900;gfx906:xnack-;gfx908:xnack-;gfx90a:xnack+;gfx90a:xnack-;gfx940;gfx941;gfx942;gfx1010;gfx1012;gfx1030;gfx1100;gfx1101;gfx1102"
105
107
] ;
106
108
} ;
107
109
apps . llama-server = {
You can’t perform that action at this time.
0 commit comments