fix vulkan on windows
This commit is contained in:
parent
ace3d10468
commit
e61c329435
|
|
@ -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 {
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue