From aa01a4c1c2fa2e823df1a868de100075eb6d6067 Mon Sep 17 00:00:00 2001 From: Lumpiasty Date: Wed, 15 Jul 2026 21:30:08 +0200 Subject: [PATCH] tmp: probe CONCAT shapes --- ggml/src/ggml-vulkan/ggml-vulkan.cpp | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/ggml/src/ggml-vulkan/ggml-vulkan.cpp b/ggml/src/ggml-vulkan/ggml-vulkan.cpp index dcdfc3ac9..84f31a324 100644 --- a/ggml/src/ggml-vulkan/ggml-vulkan.cpp +++ b/ggml/src/ggml-vulkan/ggml-vulkan.cpp @@ -12217,6 +12217,21 @@ static void ggml_vk_opt_step_sgd(ggml_backend_vk_context * ctx, vk_context& subc static void ggml_vk_concat(ggml_backend_vk_context * ctx, vk_context& subctx, const ggml_tensor * src0, const ggml_tensor * src1, ggml_tensor * dst) { int * op_params = (int *)dst->op_params; + { + static int _cc = 0; + if (_cc < 6) { + fprintf(stderr, "PROBE CONCAT axis=%d s0(ne=[%zu,%zu,%zu,%zu] nb=[%zu,%zu,%zu,%zu]) s1(ne=[%zu,%zu,%zu,%zu] nb=[%zu,%zu,%zu,%zu]) d(ne=[%zu,%zu,%zu,%zu] nb=[%zu,%zu,%zu,%zu]) type=%d\n", + op_params[0], + src0->ne[0], src0->ne[1], src0->ne[2], src0->ne[3], + src0->nb[0], src0->nb[1], src0->nb[2], src0->nb[3], + src1->ne[0], src1->ne[1], src1->ne[2], src1->ne[3], + src1->nb[0], src1->nb[1], src1->nb[2], src1->nb[3], + dst->ne[0], dst->ne[1], dst->ne[2], dst->ne[3], + dst->nb[0], dst->nb[1], dst->nb[2], dst->nb[3], src0->type); + _cc++; + } + } + const uint32_t src0_type_size = ggml_type_size(src0->type); const uint32_t src1_type_size = ggml_type_size(src1->type); const uint32_t dst_type_size = ggml_type_size(dst->type);