Skip to content

Commit a7da6ac

Browse files
committed
cast input to kernel type to improve similarity to cpu results
1 parent 4fa0331 commit a7da6ac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-vulkan/vulkan-shaders/conv2d_mm.comp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ void main() {
226226
uint32_t W_idx = OW_idx * p.s0 + KW_idx_b * p.d0 - p.p0;
227227
uint32_t src_idx =
228228
min(max(W_idx + H_idx * p.nb11 + Cin_idx_b * p.nb12 + N_idx * p.nb13, 0), p.Cin * p.N * p.W * p.H - 1);
229-
float val = src_data[src_idx];
229+
float val = A_TYPE(src_data[src_idx]);
230230
if (CRS_idx_b >= CRS || NPQ_idx >= NPQ || H_idx < 0 || H_idx >= p.H || W_idx < 0 || W_idx >= p.W) {
231231
val = 0.0;
232232
}

0 commit comments

Comments
 (0)