haraldwolff
/
whisper.cpp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

metal : wrap each operation in debug group (ggml/690)

pull/1753/head^2
Jack Mousseau 2024-01-10 06:19:19 -08:00 committed by Georgi Gerganov
parent e66a9a7806
commit a8ba1262ff
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ggml-metal.m
View File

@ -1067,6 +1067,8 @@ bool ggml_metal_graph_compute(
GGML_ASSERT(!"unsupported op");
}
[encoder pushDebugGroup:[NSString stringWithCString:ggml_op_desc(dst)]];
const int64_t ne00 = src0 ? src0->ne[0] : 0;
const int64_t ne01 = src0 ? src0->ne[1] : 0;
const int64_t ne02 = src0 ? src0->ne[2] : 0;
@ -2423,6 +2425,8 @@ bool ggml_metal_graph_compute(
GGML_ASSERT(false);
}
}
[encoder popDebugGroup];
}
if (encoder != nil) {