Upload immediately to device.
This commit is contained in:
parent
0631ea363c
commit
99c5c9a0d8
1 changed files with 1 additions and 1 deletions
|
@ -1776,7 +1776,7 @@ static struct ggml_cgraph * llama_build_graph(
|
||||||
ggml_free(ctx0);
|
ggml_free(ctx0);
|
||||||
|
|
||||||
#if defined(GGML_USE_KOMPUTE)
|
#if defined(GGML_USE_KOMPUTE)
|
||||||
if (lctx.ctx_kompute && N == 1) {
|
if (lctx.ctx_kompute) {
|
||||||
if (!ggml_vk_has_h2d_all(lctx.ctx_kompute)) {
|
if (!ggml_vk_has_h2d_all(lctx.ctx_kompute)) {
|
||||||
ggml_vk_h2d_all(lctx.ctx_kompute);
|
ggml_vk_h2d_all(lctx.ctx_kompute);
|
||||||
} else {
|
} else {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue