From a8ba1262ffb7c49a6a0d9aa023e0a764fc706484 Mon Sep 17 00:00:00 2001
From: Jack Mousseau <jmousseau@users.noreply.github.com>
Date: Wed, 10 Jan 2024 06:19:19 -0800
Subject: [PATCH] metal : wrap each operation in debug group (ggml/690)

---
 ggml-metal.m | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ggml-metal.m b/ggml-metal.m
index 6c2a8d04..16190682 100644
--- a/ggml-metal.m
+++ b/ggml-metal.m
@@ -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) {