Commit 5387ef2
1 file changed
+1
-1
lines changed- csrc/flashmask_v2/flash.h+1
- csrc/flashmask_v2/flash_bwd_kernel_sm90.h+41-38
- csrc/flashmask_v2/flash_bwd_launch_template.h+9-1
- csrc/flashmask_v2/flash_fwd_kernel_sm90.h+2-3
- csrc/flashmask_v2/flash_mask.hpp+36-19
- csrc/flashmask_v2/flash_prepare_scheduler.cu+13-2
- csrc/flashmask_v2/mainloop_bwd_sm90_tma_gmma_ws.hpp+84-106
- csrc/flashmask_v2/mainloop_fwd_sm90_tma_gmma_ws.hpp+25-28
- csrc/flashmask_v2/named_barrier.hpp+3-5
- csrc/flashmask_v2/tile_scheduler.hpp+132
0 commit comments