@@ -13698,15 +13698,114 @@ static void ggml_compute_forward_cross_entropy_loss_back(
13698
13698
13699
13699
/////////////////////////////////
13700
13700
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
+
13701
13800
static void ggml_compute_forward(struct ggml_compute_params * params, struct ggml_tensor * tensor) {
13702
13801
GGML_ASSERT(params);
13703
13802
13704
13803
if (tensor->op == GGML_OP_NONE) {
13705
13804
return;
13706
13805
}
13707
13806
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] ,
13710
13809
tensor->src[0]->name,
13711
13810
tensor->src[1]->name,
13712
13811
tensor->name);
0 commit comments