Skip to content

Commit 780e786

Browse files
cont : bump -INF block size to 64
Co-authored-by: Jeff Bolz <[email protected]>
1 parent 5088c8a commit 780e786

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test-backend-ops.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ static void init_tensor_kq_mask(ggml_tensor * tensor, float min = -1.0f, float m
150150

151151
// block size
152152
const int blck0 = 128;
153-
const int blck1 = 16;
153+
const int blck1 = 64;
154154

155155
// number of INF blocks
156156
const int n_inf_blocks = 0.1*(ne0*ne1*ne2*ne3)/(blck0*blck1);

0 commit comments

Comments
 (0)