diff --git a/csrc/punica/bgmv/bgmv_config.h b/csrc/punica/bgmv/bgmv_config.h index 95fa360f..994ac197 100644 --- 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