Skip to content

Commit 08ccee1

Browse files
authored
punica fix-bgmv-kernel-640 (#4007)
1 parent c1dc547 commit 08ccee1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

csrc/punica/bgmv/bgmv_config.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ void bgmv_kernel(out_T *__restrict__ Y, const in_T *__restrict__ X,
1414
f(in_T, out_T, W_T, narrow, 128) \
1515
f(in_T, out_T, W_T, narrow, 256) \
1616
f(in_T, out_T, W_T, narrow, 512) \
17+
f(in_T, out_T, W_T, narrow, 640) \
1718
f(in_T, out_T, W_T, narrow, 768) \
1819
f(in_T, out_T, W_T, narrow, 1024) \
1920
f(in_T, out_T, W_T, narrow, 1152) \

0 commit comments

Comments
 (0)