Skip to content

Commit 05ac2ac

Browse files
author
mike dupont
committed
adding the simple name lookup
1 parent 2824253 commit 05ac2ac

File tree

1 file changed

+101
-2
lines changed

1 file changed

+101
-2
lines changed

ggml.c

Lines changed: 101 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13698,15 +13698,114 @@ static void ggml_compute_forward_cross_entropy_loss_back(
1369813698

1369913699
/////////////////////////////////
1370013700

13701+
const char * * ggml_op_name_table [] = {
13702+
"GGML_OP_NONE",
13703+
"GGML_OP_DUP",
13704+
"GGML_OP_ADD",
13705+
"GGML_OP_ADD1",
13706+
"GGML_OP_ACC",
13707+
"GGML_OP_SUB",
13708+
"GGML_OP_MUL",
13709+
"GGML_OP_DIV",
13710+
"GGML_OP_SQR",
13711+
"GGML_OP_SQRT",
13712+
"GGML_OP_LOG",
13713+
"GGML_OP_SUM",
13714+
"GGML_OP_SUM_ROWS",
13715+
"GGML_OP_MEAN",
13716+
"GGML_OP_ARGMAX",
13717+
"GGML_OP_REPEAT",
13718+
"GGML_OP_REPEAT_BACK",
13719+
"GGML_OP_CONCAT",
13720+
"GGML_OP_SILU_BACK",
13721+
"GGML_OP_NORM",
13722+
"GGML_OP_RMS_NORM",
13723+
"GGML_OP_RMS_NORM_BACK",
13724+
"GGML_OP_GROUP_NORM",
13725+
"GGML_OP_MUL_MAT",
13726+
"GGML_OP_OUT_PROD",
13727+
"GGML_OP_SCALE",
13728+
"GGML_OP_SET",
13729+
"GGML_OP_CPY",
13730+
"GGML_OP_CONT",
13731+
"GGML_OP_RESHAPE",
13732+
"GGML_OP_VIEW",
13733+
"GGML_OP_PERMUTE",
13734+
"GGML_OP_TRANSPOSE",
13735+
"GGML_OP_GET_ROWS",
13736+
"GGML_OP_GET_ROWS_BACK",
13737+
"GGML_OP_DIAG",
13738+
"GGML_OP_DIAG_MASK_INF",
13739+
"GGML_OP_DIAG_MASK_ZERO",
13740+
"GGML_OP_SOFT_MAX",
13741+
"GGML_OP_SOFT_MAX_BACK",
13742+
"GGML_OP_ROPE",
13743+
"GGML_OP_ROPE_BACK",
13744+
"GGML_OP_ALIBI",
13745+
"GGML_OP_CLAMP",
13746+
"GGML_OP_CONV_TRANSPOSE_1D",
13747+
"GGML_OP_IM2COL",
13748+
"GGML_OP_CONV_TRANSPOSE_2D",
13749+
"GGML_OP_POOL_1D",
13750+
"GGML_OP_POOL_2D",
13751+
"GGML_OP_UPSCALE",
13752+
"GGML_OP_FLASH_ATTN",
13753+
"GGML_OP_FLASH_FF",
13754+
"GGML_OP_FLASH_ATTN_BACK",
13755+
"GGML_OP_WIN_PART",
13756+
"GGML_OP_WIN_UNPART",
13757+
"GGML_OP_GET_REL_POS",
13758+
"GGML_OP_ADD_REL_POS",
13759+
"GGML_OP_UNARY",
13760+
"GGML_OP_MAP_UNARY",
13761+
"GGML_OP_MAP_BINARY",
13762+
"GGML_OP_MAP_CUSTOM1_F32",
13763+
"GGML_OP_MAP_CUSTOM2_F32",
13764+
"GGML_OP_MAP_CUSTOM3_F32",
13765+
"GGML_OP_MAP_CUSTOM1",
13766+
"GGML_OP_MAP_CUSTOM2",
13767+
"GGML_OP_MAP_CUSTOM3",
13768+
"GGML_OP_CROSS_ENTROPY_LOSS",
13769+
"GGML_OP_CROSS_ENTROPY_LOSS_BACK",
13770+
"GGML_OP_COUNT",
13771+
};
13772+
13773+
/* enum ggml_unary_op { */
13774+
/* GGML_UNARY_OP_ABS, */
13775+
/* GGML_UNARY_OP_SGN, */
13776+
/* GGML_UNARY_OP_NEG, */
13777+
/* GGML_UNARY_OP_STEP, */
13778+
/* GGML_UNARY_OP_TANH, */
13779+
/* GGML_UNARY_OP_ELU, */
13780+
/* GGML_UNARY_OP_RELU, */
13781+
/* GGML_UNARY_OP_GELU, */
13782+
/* GGML_UNARY_OP_GELU_QUICK, */
13783+
/* GGML_UNARY_OP_SILU, */
13784+
/* GGML_UNARY_OP_LEAKY */
13785+
/* }; */
13786+
13787+
/* enum ggml_object_type { */
13788+
/* GGML_OBJECT_TENSOR, */
13789+
/* GGML_OBJECT_GRAPH, */
13790+
/* GGML_OBJECT_WORK_BUFFER */
13791+
/* }; */
13792+
13793+
/* enum ggml_log_level { */
13794+
/* GGML_LOG_LEVEL_ERROR = 2, */
13795+
/* GGML_LOG_LEVEL_WARN = 3, */
13796+
/* GGML_LOG_LEVEL_INFO = 4 */
13797+
/* }; */
13798+
13799+
1370113800
static void ggml_compute_forward(struct ggml_compute_params * params, struct ggml_tensor * tensor) {
1370213801
GGML_ASSERT(params);
1370313802

1370413803
if (tensor->op == GGML_OP_NONE) {
1370513804
return;
1370613805
}
1370713806
if (tensor->src[1]) {
13708-
fprintf(stderr, "JSON: { \"bop\":%d, \"src\":\"%s\",\"src2\":\"%s\",\"dst\":\"%s\"}\n",
13709-
tensor->op,
13807+
fprintf(stderr, "JSON: { \"bop\":%s, \"src\":\"%s\",\"src2\":\"%s\",\"dst\":\"%s\"}\n",
13808+
ggml_op_name_table[tensor->op],
1371013809
tensor->src[0]->name,
1371113810
tensor->src[1]->name,
1371213811
tensor->name);

0 commit comments

Comments
 (0)