Jack Mousseau commited on
Commit
b5e360f
·
unverified ·
1 Parent(s): ded2b1a

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

Browse files
Files changed (1) hide show
  1. ggml-metal.m +4 -0
ggml-metal.m CHANGED
@@ -1067,6 +1067,8 @@ bool ggml_metal_graph_compute(
1067
  GGML_ASSERT(!"unsupported op");
1068
  }
1069
 
 
 
1070
  const int64_t ne00 = src0 ? src0->ne[0] : 0;
1071
  const int64_t ne01 = src0 ? src0->ne[1] : 0;
1072
  const int64_t ne02 = src0 ? src0->ne[2] : 0;
@@ -2423,6 +2425,8 @@ bool ggml_metal_graph_compute(
2423
  GGML_ASSERT(false);
2424
  }
2425
  }
 
 
2426
  }
2427
 
2428
  if (encoder != nil) {
 
1067
  GGML_ASSERT(!"unsupported op");
1068
  }
1069
 
1070
+ [encoder pushDebugGroup:[NSString stringWithCString:ggml_op_desc(dst)]];
1071
+
1072
  const int64_t ne00 = src0 ? src0->ne[0] : 0;
1073
  const int64_t ne01 = src0 ? src0->ne[1] : 0;
1074
  const int64_t ne02 = src0 ? src0->ne[2] : 0;
 
2425
  GGML_ASSERT(false);
2426
  }
2427
  }
2428
+
2429
+ [encoder popDebugGroup];
2430
  }
2431
 
2432
  if (encoder != nil) {