From f0913a8fc0d49cd93bcbbc5ea6c7f8c8fff350ce Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Wed, 27 Mar 2024 13:10:13 +0200 Subject: [PATCH] examples : more CUDA leftovers (#0) --- examples/magika/CMakeLists.txt | 4 ++-- examples/python/ggml/__init__.pyi | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/magika/CMakeLists.txt b/examples/magika/CMakeLists.txt index 5543237b4..b0529f6ff 100644 --- 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 index b08ed40f9..0c2770982 100644 --- 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);"""