Skip to content

Commit 5e7ceec

Browse files
fix off-by-one
1 parent 2bc0bfb commit 5e7ceec

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

examples/lookup/lookup-create.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ int main(int argc, char ** argv){
6262
for (int i = i_start; i < inp_size; ++i) {
6363
const int ngram_start = i - ngram_size;
6464
uint64_t ngram = inp_data[ngram_start];
65-
for (int j = ngram_start; j < ngram_start + ngram_size; ++j) {
65+
for (int j = ngram_start+1; j < ngram_start + ngram_size; ++j) {
6666
const uint64_t ngram_part = inp_data[j];
6767
ngram <<= 16;
6868
ngram |= ngram_part;

examples/lookup/lookup.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -284,7 +284,7 @@ int main(int argc, char ** argv){
284284

285285
const int static_ngram_start = inp_size-2 + draft.size()-1;
286286
uint64_t static_ngram = get_token(inp, draft, static_ngram_start);
287-
for (int j = static_ngram_start; j < static_ngram_start + 2; ++j) {
287+
for (int j = static_ngram_start+1; j < static_ngram_start + 2; ++j) {
288288
const uint64_t ngram_part = get_token(inp, draft, j);
289289
static_ngram <<= 16;
290290
static_ngram |= ngram_part;
@@ -304,7 +304,7 @@ int main(int argc, char ** argv){
304304

305305
const int ngram_start = inp_size-ngram_size + draft.size()-1;
306306
uint64_t ngram = get_token(inp, draft, ngram_start);
307-
for (int j = ngram_start; j < ngram_start + ngram_size; ++j) {
307+
for (int j = ngram_start+1; j < ngram_start + ngram_size; ++j) {
308308
const uint64_t ngram_part = get_token(inp, draft, j);
309309
ngram <<= 16;
310310
ngram |= ngram_part;

0 commit comments

Comments
 (0)