examples : more CUDA leftovers (#0)

ggerganov · Mar 27, 2024 · f0913a8 · f0913a8
1 parent fe3158e
commit f0913a8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/examples/magika/CMakeLists.txt b/examples/magika/CMakeLists.txt
@@ -8,8 +8,8 @@ target_link_libraries(${TEST_TARGET} PRIVATE ggml common common-ggml)
 #
 # For GPU offloading
 
-if (GGML_CUBLAS)
- add_compile_definitions(GGML_USE_CUBLAS)
+if (GGML_CUDA)
+ add_compile_definitions(GGML_USE_CUDA)
 endif()
 
 if (GGML_CLBLAST)

diff --git a/examples/python/ggml/__init__.pyi b/examples/python/ggml/__init__.pyi
@@ -568,8 +568,8 @@ class lib:
  def ggml_cpu_has_clblast() -> int:
  """ GGML_API int ggml_cpu_has_clblast (void);"""
  ...
- def ggml_cpu_has_cublas() -> int:
- """ GGML_API int ggml_cpu_has_cublas (void);"""
+ def ggml_cpu_has_cuda() -> int:
+ """ GGML_API int ggml_cpu_has_cuda  (void);"""
  ...
  def ggml_cpu_has_f16c() -> int:
  """ GGML_API int ggml_cpu_has_f16c (void);"""
@@ -967,8 +967,8 @@ class lib:
  def ggml_init(params: ffi.CData) -> ffi.CData:
  """ GGML_API struct ggml_context * ggml_init(struct ggml_init_params params);"""
  ...
- def ggml_init_cublas() -> None:
- """GGML_API void ggml_init_cublas(void);"""
+ def ggml_init_cuda() -> None:
+ """GGML_API void ggml_init_cuda(void);"""
  ...
  def ggml_internal_get_type_traits(type: int) -> ffi.CData:
  """ ggml_type_traits_t ggml_internal_get_type_traits(enum ggml_type type);"""