Add Vulkan device validation to Net::load_param_bin

Tencent · Oct 17, 2024 · a326f08 · a326f08
1 parent c15d345
commit a326f08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/net.cpp b/src/net.cpp
@@ -1677,7 +1677,7 @@ int Net::load_param_bin(const DataReader& dr)
     if (opt.use_vulkan_compute)
     {
         if (!d->vkdev) d->vkdev = get_gpu_device();
-        if (!d->vkdev) opt.use_vulkan_compute = false; // no vulkan device, fallback to cpu
+        if (!d->vkdev || !d->vkdev->is_valid()) opt.use_vulkan_compute = false; // no valid vulkan device, fallback to cpu
     }
     if (opt.use_vulkan_compute)
     {