diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c index 500cb6b72f..9a5a48e531 100644 --- a/ggml/src/ggml.c +++ b/ggml/src/ggml.c @@ -6905,7 +6905,10 @@ void ggml_build_backward_expand( // inplace operations are currently not supported GGML_ASSERT(!node->view_src || node->op == GGML_OP_CPY || node->op == GGML_OP_VIEW || - node->op == GGML_OP_RESHAPE || node->op == GGML_OP_PERMUTE || node->op == GGML_OP_TRANSPOSE); + node->op == GGML_OP_RESHAPE || node->op == GGML_OP_PERMUTE || node->op == GGML_OP_TRANSPOSE || + node->op == GGML_OP_ADD || node->op == GGML_OP_ACC || node->op == GGML_OP_DUP || node->op == GGML_OP_SCALE || + node->op == GGML_OP_SUB || node->op == GGML_OP_MUL || node->op == GGML_OP_DIV || + node->op == GGML_OP_OPT_STEP_ADAMW || node->op == GGML_OP_OPT_STEP_SGD); const size_t ihash = ggml_hash_find(&cgraph->visited_hash_set, node); GGML_ASSERT(ihash != GGML_HASHSET_FULL);