llama.cpp updated

mgonzs13 · Mar 26, 2024 · e61fd0c · e61fd0c
1 parent 012aa18
commit e61fd0c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -33,8 +33,8 @@ $ colcon build
 To run llama_ros with CUDA, you have to install the [CUDA Toolkit](https://developer.nvidia.com/cuda-toolkit) and the following lines in the [CMakeLists.txt](llama_ros/CMakeLists.txt) of llama_ros package must be uncommented:
 
 ```
-option(LLAMA_CUBLAS "llama: use cuBLAS" ON)
-add_compile_definitions(GGML_USE_CUBLAS)
+option(LLAMA_CUDA "llama: use CUDA" ON)
+add_compile_definitions(GGML_USE_CUDA)
 ```
 
 ## Usage

diff --git a/llama_ros/CMakeLists.txt b/llama_ros/CMakeLists.txt
@@ -6,8 +6,8 @@ if(CMAKE_COMPILER_IS_GNUCXX OR CMAKE_CXX_COMPILER_ID MATCHES "Clang")
 endif()
 
 # cuBLAS
-# option(LLAMA_CUBLAS "llama: use cuBLAS" ON)
-# add_compile_definitions(GGML_USE_CUBLAS)
+# option(LLAMA_CUDA "llama: use CUDA" ON)
+# add_compile_definitions(GGML_USE_CUDA)
 
 # find dependencies
 find_package(ament_cmake REQUIRED)

diff --git a/llama_ros/llama_cpp b/llama_ros/llama_cpp