Skip to content

Commit 1aafb78

Browse files
committed
CANN: add op_params match
1 parent 3d404b4 commit 1aafb78

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

ggml/src/ggml-cann/ggml-cann.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2257,9 +2257,8 @@ static bool ggml_graph_node_has_matching_properties(
22572257
}
22582258
}
22592259

2260-
if (node->op == GGML_OP_SCALE &&
2261-
memcmp(graph_node_properties->op_params, node->op_params, GGML_MAX_OP_PARAMS) != 0) {
2262-
return false;
2260+
if (node->op == GGML_OP_SCALE || node->op == GGML_OP_UNARY || node->op == GGML_OP_GLU) {
2261+
return memcmp(graph_node_properties->op_params, node->op_params, GGML_MAX_OP_PARAMS) == 0;
22632262
}
22642263
return true;
22652264
}

0 commit comments

Comments
 (0)