Skip to content

Commit 218f2d7

Browse files
committed
fix validation error for large im2col
1 parent 64a367b commit 218f2d7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8620,6 +8620,10 @@ static void ggml_vk_op_f32(ggml_backend_vk_context * ctx, vk_context& subctx, co
86208620

86218621
ggml_vk_dispatch_pipeline(ctx, subctx, pipeline, { vk_subbuffer{ d_X, x_buf_offset, x_sz }, vk_subbuffer{ d_Y, y_buf_offset, y_sz }, subbuf_z, vk_subbuffer{ d_D, d_buf_offset, d_sz } }, pc, elements);
86228622
} else if (op == GGML_OP_IM2COL || op == GGML_OP_IM2COL_3D) {
8623+
if (ctx->device->shader_int64 && ctx->device->buffer_device_address) {
8624+
// buffer device address path doesn't use dst buffer
8625+
d_sz = 1;
8626+
}
86238627
// im2col uses only src1 and dst buffers
86248628
ggml_vk_dispatch_pipeline(ctx, subctx, pipeline, { vk_subbuffer{ d_Y, y_buf_offset, y_sz }, vk_subbuffer{ d_D, d_buf_offset, d_sz } }, pc, elements);
86258629
} else if (op == GGML_OP_COUNT_EQUAL) {

0 commit comments

Comments
 (0)