Fix some nodes are not checked with GGML_VULKAN_CHECK_RESULTS enabled.

ggerganov · Sep 8, 2024 · 9dc0223 · 9dc0223
1 parent 28506e5
commit 9dc0223
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ggml/src/ggml-vulkan.cpp b/ggml/src/ggml-vulkan.cpp
@@ -6478,14 +6478,18 @@ GGML_CALL static ggml_status ggml_backend_vk_graph_compute(ggml_backend_t backen
         }
 
         bool submit = (submitted_nodes >= submit_count) || (i == last_node);
+
+
         bool enqueued = ggml_vk_build_graph(ctx, cgraph->nodes[i], i, cgraph->nodes[submit_node_idx], submit_node_idx, false, i == last_node, submit);
 
         if (enqueued) {
             ++submitted_nodes;
 
+#ifndef GGML_VULKAN_CHECK_RESULTS
             if (first_node_in_batch) {
                 first_node_in_batch = false;
             }
+#endif
         }
 
         if (submit) {