Add Vulkan to CMake pkg

This commit is contained in:
Mason M 2024-06-24 15:42:49 -03:00
parent d62e4aaa02
commit 4eadfb11ee

View file

@ -7,6 +7,11 @@ set(LLAMA_CUDA @LLAMA_CUDA@)
set(LLAMA_METAL @LLAMA_METAL@) set(LLAMA_METAL @LLAMA_METAL@)
set(LLAMA_HIPBLAS @LLAMA_HIPBLAS@) set(LLAMA_HIPBLAS @LLAMA_HIPBLAS@)
set(LLAMA_ACCELERATE @LLAMA_ACCELERATE@) set(LLAMA_ACCELERATE @LLAMA_ACCELERATE@)
set(LLAMA_VULKAN @LLAMA_VULKAN@)
set(LLAMA_VULKAN_CHECK_RESULTS @LLAMA_VULKAN_CHECK_RESULTS@)
set(LLAMA_VULKAN_DEBUG @LLAMA_VULKAN_DEBUG@)
set(LLAMA_VULKAN_MEMORY_DEBUG @LLAMA_VULKAN_MEMORY_DEBUG@)
set(LLAMA_VULKAN_VALIDATE @LLAMA_VULKAN_VALIDATE@)
@PACKAGE_INIT@ @PACKAGE_INIT@
@ -35,6 +40,10 @@ if (LLAMA_METAL)
find_library(METALKIT_FRAMEWORK MetalKit REQUIRED) find_library(METALKIT_FRAMEWORK MetalKit REQUIRED)
endif() endif()
if (LLAMA_VULKAN)
find_package(Vulkan)
endif()
if (LLAMA_HIPBLAS) if (LLAMA_HIPBLAS)
find_package(hip REQUIRED) find_package(hip REQUIRED)
find_package(hipblas REQUIRED) find_package(hipblas REQUIRED)