Skip to content

Commit 3c81e10

Browse files
fix off-by-one
1 parent c8287f6 commit 3c81e10

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
@@ -283,7 +283,7 @@ int main(int argc, char ** argv){
283283

284284
const int static_ngram_start = inp_size-2 + draft.size()-1;
285285
uint64_t static_ngram = get_token(inp, draft, static_ngram_start);
286-
for (int j = static_ngram_start; j < static_ngram_start + 2; ++j) {
286+
for (int j = static_ngram_start+1; j < static_ngram_start + 2; ++j) {
287287
const uint64_t ngram_part = get_token(inp, draft, j);
288288
static_ngram <<= 16;
289289
static_ngram |= ngram_part;
@@ -303,7 +303,7 @@ int main(int argc, char ** argv){
303303

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

0 commit comments

Comments
 (0)