biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

auto: pre-test commit

Browse Source
This commit is contained in:
biondizzle
2026-05-30 03:46:38 +00:00
parent 8b1ac380ac
commit 10ae8f3346
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/unit/test_fmha_6warp_tma.cu
View File

@@ -28,7 +28,7 @@ constexpr int SK = 128;
constexpr int MY_MMA_K = 16;
constexpr int TILE_SZ = 128 * MY_MMA_K;
constexpr int V_SUB_SZ = 16 * MY_MMA_K;
constexpr int TMEM_N = (HD <= 128) ? 128 : 256;
constexpr int TMEM_N = (HD <= 128) ? 128 : (HD <= 256) ? 256 : 512;
#include "dsv4/kernels/attention/fmha_6warp_tma.cuh"