Skip to content

Commit acb17f6

Browse files
committed
Rename cuBLAS to CUDA
1 parent d39fc55 commit acb17f6

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ A PowerShell automation to rebuild [llama.cpp](https://github.com/ggerganov/llam
1111

1212
## BLAS support
1313

14-
This script currently supports `OpenBLAS` for CPU BLAS acceleration and `cuBLAS` for NVIDIA GPU BLAS acceleration.
14+
This script currently supports `OpenBLAS` for CPU BLAS acceleration and `CUDA` for NVIDIA GPU BLAS acceleration.
1515

1616
## Installation
1717

@@ -241,7 +241,7 @@ Every time there is a new release of [llama.cpp](https://github.com/ggerganov/ll
241241
| `./rebuild_llama.cpp.ps1` | Automatically detects best BLAS acceleration |
242242
| `./rebuild_llama.cpp.ps1 -blasAccelerator "OFF"` | Without any BLAS acceleration |
243243
| `./rebuild_llama.cpp.ps1 -blasAccelerator "OpenBLAS"` | With CPU BLAS acceleration |
244-
| `./rebuild_llama.cpp.ps1 -blasAccelerator "cuBLAS"` | With NVIDIA GPU BLAS acceleration |
244+
| `./rebuild_llama.cpp.ps1 -blasAccelerator "CUDA"` | With NVIDIA GPU BLAS acceleration |
245245

246246
### Build a specific version of llama.cpp
247247

rebuild_llama.cpp.ps1

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ Automatically rebuild llama.cpp for a Windows environment.
88
This script automatically rebuilds llama.cpp for a Windows environment.
99
1010
.PARAMETER blasAccelerator
11-
Specifies the BLAS accelerator, supported values are: "OpenBLAS", "cuBLAS", "OFF"
11+
Specifies the BLAS accelerator, supported values are: "OpenBLAS", "CUDA", "OFF"
1212
1313
.PARAMETER version
1414
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.
2020
.\rebuild_llama.cpp.ps1 -blasAccelerator "OpenBLAS"
2121
2222
.EXAMPLE
23-
.\rebuild_llama.cpp.ps1 -blasAccelerator "cuBLAS" -version "master-4e7464e"
23+
.\rebuild_llama.cpp.ps1 -blasAccelerator "CUDA" -version "master-4e7464e"
2424
#>
2525

2626
Param (
27-
[ValidateSet("OpenBLAS", "cuBLAS", "OFF")]
27+
[ValidateSet("OpenBLAS", "CUDA", "OFF")]
2828
[String]
2929
$blasAccelerator,
3030

@@ -62,7 +62,7 @@ if (!$blasAccelerator) {
6262
if ((Get-Command "nvidia-smi" -ErrorAction SilentlyContinue) -and
6363
(Get-Command "nvcc" -ErrorAction SilentlyContinue)) {
6464

65-
$blasAccelerator = "cuBLAS"
65+
$blasAccelerator = "CUDA"
6666
}
6767
}
6868

@@ -149,7 +149,7 @@ switch ($blasAccelerator) {
149149
..
150150
}
151151

152-
"cuBLAS" {
152+
"CUDA" {
153153
cmake `
154154
-DLLAMA_CUDA=ON `
155155
-DLLAMA_CCACHE=OFF `

0 commit comments

Comments
 (0)