fix vulkan on windows

This commit is contained in:
pufferffish 2024-09-23 18:43:16 +08:00
parent ace3d10468
commit e61c329435
2 changed files with 21 additions and 0 deletions

View File

@ -40,11 +40,16 @@ var OneapiGlobs = []string{
"c:\\Windows\\System32\\DriverStore\\FileRepository\\*\\ze_intel_gpu64.dll",
}
var VulkanGlobs = []string{
"c:\\Windows\\System32\\vulkan-1.dll",
}
var (
CudartMgmtName = "cudart64_*.dll"
NvcudaMgmtName = "nvcuda.dll"
NvmlMgmtName = "nvml.dll"
OneapiMgmtName = "ze_intel_gpu64.dll"
VulkanMgmtName = "vulkan-1.dll"
)
func FindLibCapLibs() []string {

View File

@ -412,6 +412,21 @@ function build_rocm() {
}
}
function build_vulkan() {
if (-not "${env:OLLAMA_SKIP_VULKAN_GENERATE}") {
init_vars
$script:buildDir="../build/windows/${script:ARCH}/vulkan"
$script:distDir="$script:DIST_BASE\vulkan"
$script:cmakeDefs += @("-A", "x64", "-DLLAMA_VULKAN=1")
write-host "Building Vulkan"
build
sign
install
} else {
write-host "Skipping Vulkan generation step"
}
}
init_vars
if ($($args.count) -eq 0) {
git_module_setup
@ -426,6 +441,7 @@ if ($($args.count) -eq 0) {
build_cuda
build_oneapi
build_rocm
build_vulkan
}
cleanup