metal : put encoder debug group behind a define (llama/4873)

pull/1753/head^2
Paul Tsochantaris 2024-01-11 14:31:52 +00:00 committed by Georgi Gerganov
parent a0a64a19dd
commit 9fa34d79ec
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 4 additions and 0 deletions

View File

@ -1067,7 +1067,9 @@ bool ggml_metal_graph_compute(
GGML_ASSERT(!"unsupported op");
}
#ifndef GGML_METAL_NDEBUG
[encoder pushDebugGroup:[NSString stringWithCString:ggml_op_desc(dst) encoding:NSUTF8StringEncoding]];
#endif
const int64_t ne00 = src0 ? src0->ne[0] : 0;
const int64_t ne01 = src0 ? src0->ne[1] : 0;
@ -2426,7 +2428,9 @@ bool ggml_metal_graph_compute(
}
}
#ifndef GGML_METAL_NDEBUG
[encoder popDebugGroup];
#endif
}
if (encoder != nil) {