Skip to content

Commit acc31b1

Browse files
ruixiang63tnhnyzc
andcommitted
eagle3: set sync when get features from target
Co-authored-by: tnhnyzc <115956684+tnhnyzc@users.noreply.github.com>
1 parent cfd8e7a commit acc31b1

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/llama-context.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4084,5 +4084,7 @@ void llama_set_output_layer_inp(struct llama_context * ctx, uint32_t layer_id, b
40844084
}
40854085

40864086
float * llama_get_output_layer_inp(struct llama_context * ctx, uint32_t layer_id) {
4087+
ctx->synchronize();
4088+
40874089
return ctx->get_output_layer_inp(layer_id);
40884090
}

0 commit comments

Comments
 (0)