[vulkan] Fixed query pool invalid usage (#5717)

* Reset query pools after command buffer execution

* Removed clang extension syntax

* Fixed compilation

* Fixed reset query pool

* Disable validation

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

c_api/src/c_api_test_utils.cpp

@@ -60,7 +60,7 @@ TiNdarrayAndMem make_ndarray(TiRuntime runtime,
     alloc_size = 1;
 
   } else {
-    assert(false);
+    TI_ASSERT(false);
   }
 
   for (int i = 0; i < arr_dims; i++) {
@@ -95,15 +95,17 @@ TiNdarrayAndMem make_ndarray(TiRuntime runtime,
     e_shape.dims[i] = element_shape[i];
   }
 
-  TiNdArray arg_array = {.memory = res.memory_,
-                         .shape = shape,
-                         .elem_shape = e_shape,
-                         .elem_type = dtype};
+  TiNdArray arg_array{};
+  arg_array.memory = res.memory_;
+  arg_array.shape = shape;
+  arg_array.elem_shape = e_shape;
+  arg_array.elem_type = dtype;
 
-  TiArgumentValue arg_value = {.ndarray = arg_array};
+  TiArgumentValue arg_value{};
+  arg_value.ndarray = arg_array;
 
-  res.arg_ = {.type = TiArgumentType::TI_ARGUMENT_TYPE_NDARRAY,
-              .value = arg_value};
+  res.arg_.type = TiArgumentType::TI_ARGUMENT_TYPE_NDARRAY;
+  res.arg_.value = arg_value;
 
   return res;
 }

taichi/python/export_misc.cpp

@@ -140,7 +140,11 @@ void export_misc(py::module &m) {
   m.def("pop_python_print_buffer", []() { return py_cout.pop_content(); });
   m.def("toggle_python_print_buffer", [](bool opt) { py_cout.enabled = opt; });
   m.def("with_cuda", is_cuda_api_available);
+#ifdef TI_WITH_METAL
   m.def("with_metal", taichi::lang::metal::is_metal_api_available);
+#else
+  m.def("with_metal", []() { return false; });
+#endif
 #ifdef TI_WITH_OPENGL
   m.def("with_opengl", taichi::lang::opengl::is_opengl_api_available,
         py::arg("use_gles") = false);

taichi/rhi/vulkan/vulkan_device.cpp

@@ -782,6 +782,7 @@ VulkanCommandList::VulkanCommandList(VulkanDevice *ti_device,
   info.flags = VK_COMMAND_BUFFER_USAGE_SIMULTANEOUS_USE_BIT;
 
   vkBeginCommandBuffer(buffer->buffer, &info);
+  vkCmdResetQueryPool(buffer->buffer, query_pool_->query_pool, 0, 2);
   vkCmdWriteTimestamp(buffer->buffer, VK_PIPELINE_STAGE_ALL_COMMANDS_BIT,
                       query_pool_->query_pool, 0);
 }