summaryrefslogtreecommitdiff
path: root/examples/llama-bench/llama-bench.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'examples/llama-bench/llama-bench.cpp')
-rwxr-xr-xexamples/llama-bench/llama-bench.cpp4
1 files changed, 1 insertions, 3 deletions
diff --git a/examples/llama-bench/llama-bench.cpp b/examples/llama-bench/llama-bench.cpp
index 36057bfc..7a281158 100755
--- a/examples/llama-bench/llama-bench.cpp
+++ b/examples/llama-bench/llama-bench.cpp
@@ -18,9 +18,7 @@
#include "llama.h"
#include "common.h"
#include "build-info.h"
-#ifdef GGML_USE_CUBLAS
#include "ggml-cuda.h"
-#endif
// utils
static uint64_t get_time_ns() {
@@ -504,7 +502,7 @@ struct test {
static std::string get_backend() {
if (cuda) {
- return "CUDA";
+ return GGML_CUDA_NAME;
}
if (opencl) {
return "OpenCL";