Skip to content

Commit ef31998

Browse files
authored
Update ggml_extend.hpp
1 parent befa07e commit ef31998

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

ggml_extend.hpp

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1054,13 +1054,10 @@ struct GGMLRunner {
10541054
compute_allocr = ggml_gallocr_new_n_multi(bufts, 2);
10551055

10561056
int n_nodes = ggml_graph_n_nodes(gf);
1057-
1058-
int total_n_half = n_nodes / 2;
1059-
10601057
int * node_buffer_ids = (int *)calloc(n_nodes, sizeof(int));
10611058

10621059
for (int i = 0; i < n_nodes; i++) {
1063-
node_buffer_ids[i] = i < total_n_half ? 0 : 1;
1060+
node_buffer_ids[i] = i % 2;
10641061
}
10651062

10661063
if (!ggml_gallocr_reserve_n(compute_allocr, gf, node_buffer_ids, NULL)) {

0 commit comments

Comments
 (0)