Fixed vulkan device capability settings

taichi-dev · Jul 12, 2022 · b149436 · b149436
1 parent c6b3d8a
commit b149436
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/c_api/src/taichi_vulkan_impl.cpp b/c_api/src/taichi_vulkan_impl.cpp
@@ -28,11 +28,22 @@ VulkanRuntimeImported::Workaround::Workaround(
   taichi::lang::vulkan::VulkanLoader::instance().load_device(params.device);
   vk_device.set_cap(taichi::lang::DeviceCapability::vk_api_version,
                     api_version);
+
+  vk_device.set_cap(taichi::lang::DeviceCapability::spirv_version, 0x10000);
+  if (api_version >= VK_API_VERSION_1_3) {
+    vk_device.set_cap(taichi::lang::DeviceCapability::spirv_version, 0x10500);
+  } else if (api_version >= VK_API_VERSION_1_2) {
+    vk_device.set_cap(taichi::lang::DeviceCapability::spirv_version, 0x10500);
+  } else if (api_version >= VK_API_VERSION_1_1) {
+    vk_device.set_cap(taichi::lang::DeviceCapability::spirv_version, 0x10300);
+  }
+
   if (api_version > VK_API_VERSION_1_0) {
     vk_device.set_cap(
         taichi::lang::DeviceCapability::spirv_has_physical_storage_buffer,
         true);
   }
+
   vk_device.init_vulkan_structs(
       const_cast<taichi::lang::vulkan::VulkanDevice::Params &>(params));
 }