File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -256,8 +256,6 @@ static std::vector<int> ggml_metal_graph_optimize_reorder(const std::vector<node
256256
257257 // perform reorders only across these types of ops
258258 // can be expanded when needed
259- // IMPORTANT: do not add ops such as GGML_OP_CPY or GGML_OP_SET_ROWS
260- // the dependencies from such ops are not always represented in the graph
261259 const auto & h_safe = [](ggml_op op) {
262260 switch (op) {
263261 case GGML_OP_MUL_MAT:
@@ -273,6 +271,8 @@ static std::vector<int> ggml_metal_graph_optimize_reorder(const std::vector<node
273271 case GGML_OP_GLU:
274272 case GGML_OP_SCALE:
275273 case GGML_OP_GET_ROWS:
274+ case GGML_OP_CPY:
275+ case GGML_OP_SET_ROWS:
276276 return true ;
277277 default :
278278 return ggml_op_is_empty (op);
You can’t perform that action at this time.
0 commit comments