Skip to content

Commit cbd59d4

Browse files
use float since HIP doesn't support conversion from double to hip_bfloat16
1 parent 8346185 commit cbd59d4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-cuda/win.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ __global__ static void win_part_kernel(
138138
if (py*p.w + i2 >= p.ne2 || px*p.w + i1 >= p.ne1) {
139139
for (int i0 = threadIdx.x; i0 < p.C; i0 += blockDim.x) {
140140
char * dp = (char *)dst + i3*nb3 + i2*nb2 + i1*nb1 + i0*nb0;
141-
*((T *) dp) = 0.0;
141+
*((T *) dp) = 0.0f;
142142
}
143143
return;
144144
}

0 commit comments

Comments
 (0)