Skip to content

Commit e500e8e

Browse files
committed
compat: fix sparse staging write async arguments on b9411
1 parent b729002 commit e500e8e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

ggml/src/ggml-vulkan/ggml-vulkan.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7771,7 +7771,7 @@ static void ggml_vk_buffer_write_2d(vk_buffer& dst, size_t offset, const void *
77717771
// Single-shot transfer
77727772
vk_context subctx = ggml_vk_create_temporary_context(dst->device->transfer_queue.cmd_pool);
77737773
ggml_vk_ctx_begin(dst->device, subctx);
7774-
bool ret = ggml_vk_buffer_write_2d_async(subctx, dst, offset, src, 0, width, 1, true);
7774+
bool ret = ggml_vk_buffer_write_2d_async(subctx, dst, offset, src, width, width, width, 1, true);
77757775
GGML_ASSERT(ret);
77767776
ggml_vk_ctx_end(subctx);
77777777

@@ -7813,7 +7813,7 @@ static void ggml_vk_buffer_write_2d(vk_buffer& dst, size_t offset, const void *
78137813
const size_t chunk = std::min(remaining, dst->device->sync_staging->size);
78147814

78157815
ggml_vk_ctx_begin(dst->device, subctx);
7816-
bool ret = ggml_vk_buffer_write_2d_async(subctx, dst, dst_off, src_ptr, 0, chunk, 1, true);
7816+
bool ret = ggml_vk_buffer_write_2d_async(subctx, dst, dst_off, src_ptr, chunk, chunk, chunk, 1, true);
78177817
GGML_ASSERT(ret);
78187818
ggml_vk_ctx_end(subctx);
78197819

0 commit comments

Comments
 (0)