Skip to content

Commit 89f2a4b

Browse files
Update llama-context.cpp
Cosmo STL doesn't have std:fill. Signed-off-by: Brad Hutchings <[email protected]>
1 parent 07a026e commit 89f2a4b

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

src/llama-context.cpp

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1583,7 +1583,15 @@ int32_t llama_context::output_reserve(int32_t n_outputs) {
15831583
embd = has_embd ? output_base + logits_size : nullptr;
15841584

15851585
// set all ids as invalid (negative)
1586-
std::fill(output_ids.begin(), output_ids.end(), -1);
1586+
// llama-server-one START
1587+
#ifndef COSMOCC
1588+
std::fill(output_ids.begin(), output_ids.end(), -1);
1589+
#else
1590+
for (auto iii = output_ids.begin(); iii != output_ids.end(); iii++) {
1591+
*iii = -1;
1592+
}
1593+
#endif
1594+
// llama-server-one END
15871595

15881596
ggml_backend_buffer_clear(buf_output.get(), 0);
15891597

@@ -1623,7 +1631,17 @@ void llama_context::output_reorder() {
16231631
}
16241632
}
16251633
}
1634+
1635+
// llama-server-one START
1636+
#ifndef COSMOCC
16261637
std::fill(output_ids.begin(), output_ids.end(), -1);
1638+
#else
1639+
for (auto iii = output_ids.begin(); iii != output_ids.end(); iii++) {
1640+
*iii = -1;
1641+
}
1642+
#endif
1643+
// llama-server-one END
1644+
16271645
for (int32_t i = 0; i < n_outputs; ++i) {
16281646
output_ids[out_ids[i]] = i;
16291647
}

0 commit comments

Comments
 (0)