Spaces:
Sleeping
Sleeping
Jack Mousseau
commited on
metal : wrap each operation in debug group (ggml/690)
Browse files- 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) {
|