| 1767 | static bool GGML_OP_HAS_FINALIZE[GGML_OP_COUNT] = { 0 }; |
| 1768 | |
| 1769 | static void ggml_setup_op_has_task_pass(void) { |
| 1770 | { // INIT |
| 1771 | bool * p = GGML_OP_HAS_INIT; |
| 1772 | |
| 1773 | p[GGML_OP_ACC ] = true; |
| 1774 | p[GGML_OP_MUL_MAT ] = true; |
| 1775 | p[GGML_OP_MUL_MAT_SPARSE ] = true; |
| 1776 | p[GGML_OP_AXPY ] = true; |
| 1777 | p[GGML_OP_OUT_PROD ] = true; |
| 1778 | p[GGML_OP_SET ] = true; |
| 1779 | p[GGML_OP_GET_ROWS_BACK ] = true; |
| 1780 | p[GGML_OP_DIAG_MASK_INF ] = true; |
| 1781 | p[GGML_OP_DIAG_MASK_ZERO ] = true; |
| 1782 | p[GGML_OP_CONV_TRANSPOSE_1D ] = true; |
| 1783 | p[GGML_OP_CONV_TRANSPOSE_2D ] = true; |
| 1784 | p[GGML_OP_FLASH_ATTN_BACK ] = true; |
| 1785 | p[GGML_OP_CROSS_ENTROPY_LOSS ] = true; |
| 1786 | p[GGML_OP_ADD_REL_POS ] = true; |
| 1787 | } |
| 1788 | |
| 1789 | { // FINALIZE |
| 1790 | bool * p = GGML_OP_HAS_FINALIZE; |
| 1791 | |
| 1792 | p[GGML_OP_CROSS_ENTROPY_LOSS ] = true; |
| 1793 | } |
| 1794 | } |
| 1795 | |
| 1796 | // |
| 1797 | // ggml context |