Lumpiasty/llama.cpp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tmp: probe CONCAT shapes

Browse Source
This commit is contained in:
Lumpiasty
2026-07-15 21:30:08 +02:00
parent f4b0af3f1e
commit aa01a4c1c2
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ggml/src/ggml-vulkan/ggml-vulkan.cpp
+15
View File
@@ -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) { 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; 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 src0_type_size = ggml_type_size(src0->type);
const uint32_t src1_type_size = ggml_type_size(src1->type); const uint32_t src1_type_size = ggml_type_size(src1->type);
const uint32_t dst_type_size = ggml_type_size(dst->type); const uint32_t dst_type_size = ggml_type_size(dst->type);