@@ -8,7 +8,7 @@ Automatically rebuild llama.cpp for a Windows environment.
8
8
This script automatically rebuilds llama.cpp for a Windows environment.
9
9
10
10
. PARAMETER blasAccelerator
11
- Specifies the BLAS accelerator, supported values are: "OpenBLAS", "cuBLAS ", "OFF"
11
+ Specifies the BLAS accelerator, supported values are: "OpenBLAS", "CUDA ", "OFF"
12
12
13
13
. PARAMETER version
14
14
Specifies a llama.cpp commit or tag to checkout a specific version.
@@ -20,11 +20,11 @@ Specifies a llama.cpp commit or tag to checkout a specific version.
20
20
.\rebuild_llama.cpp.ps1 -blasAccelerator "OpenBLAS"
21
21
22
22
. EXAMPLE
23
- .\rebuild_llama.cpp.ps1 -blasAccelerator "cuBLAS " -version "master-4e7464e"
23
+ .\rebuild_llama.cpp.ps1 -blasAccelerator "CUDA " -version "master-4e7464e"
24
24
#>
25
25
26
26
Param (
27
- [ValidateSet (" OpenBLAS" , " cuBLAS " , " OFF" )]
27
+ [ValidateSet (" OpenBLAS" , " CUDA " , " OFF" )]
28
28
[String ]
29
29
$blasAccelerator ,
30
30
@@ -62,7 +62,7 @@ if (!$blasAccelerator) {
62
62
if ((Get-Command " nvidia-smi" - ErrorAction SilentlyContinue) -and
63
63
(Get-Command " nvcc" - ErrorAction SilentlyContinue)) {
64
64
65
- $blasAccelerator = " cuBLAS "
65
+ $blasAccelerator = " CUDA "
66
66
}
67
67
}
68
68
@@ -149,7 +149,7 @@ switch ($blasAccelerator) {
149
149
..
150
150
}
151
151
152
- " cuBLAS " {
152
+ " CUDA " {
153
153
cmake `
154
154
- DLLAMA_CUDA= ON `
155
155
- DLLAMA_CCACHE= OFF `
0 commit comments