🐛 I don't know what I'm doing

Signed-off-by: Joe Runde <[email protected]>
IBM · Jun 14, 2024 · 47c11ae · 47c11ae
1 parent dfc39d6
commit 47c11ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/csrc/punica/bgmv/bgmv_config.h b/csrc/punica/bgmv/bgmv_config.h
@@ -172,6 +172,7 @@ void bgmv_kernel(out_T *__restrict__ Y, const in_T *__restrict__ X,
     f(in_T, out_T, W_T, 8, 64) \
     f(in_T, out_T, W_T, 16, 64) \
     f(in_T, out_T, W_T, 32, 64) \
-    f(in_T, out_T, W_T, 64, 64)
+    f(in_T, out_T, W_T, 64, 64) \
+    f(in_T, out_T, W_T, 128, 128)
 
 // clang-format on