|
|
02edff5ac7
|
D1: add KV merge test using log-sum-exp (avoids TMEM round-trip)
|
2026-05-24 22:17:24 +00:00 |
|
|
|
0f30319e06
|
Revert "D1: move O rescale atoms outside const_expr guard (match CUTLASS pattern)"
This reverts commit aaf21d8ac1.
|
2026-05-24 22:15:38 +00:00 |
|
|
|
aaf21d8ac1
|
D1: move O rescale atoms outside const_expr guard (match CUTLASS pattern)
|
2026-05-24 22:07:18 +00:00 |
|
|
|
35a3c04e8e
|
fix debug test
|
2026-05-24 22:04:51 +00:00 |
|
|
|
a391aa1fd3
|
D1: add rescale debug test
|
2026-05-24 22:04:20 +00:00 |
|
|
|
55c6903980
|
D1: fix O rescale identity tensor - use PV MMA shape not QK shape
|
2026-05-24 22:02:55 +00:00 |
|
|
|
f1aab1bfc1
|
D1: add multi-KV-tile O rescale test (s_k=256,384,512)
|
2026-05-24 22:00:42 +00:00 |
|
|
|
77b366d44b
|
Scrub B200 password from markdown files
|
2026-05-24 21:52:54 +00:00 |
|
|
|
83506e6ad2
|
Add MAY_24_26_PLAN.md: next session startup plan
|
2026-05-24 21:50:32 +00:00 |
|
|
|
9435bf9653
|
Restore NVFP4 Precision Roadmap + add O rescale gap to D1.5
|
2026-05-24 21:48:58 +00:00 |
|
|
|
03cbd8ffa6
|
Add STAGE_D2.md: Multi-query grid + head packing plan
|
2026-05-24 21:43:04 +00:00 |
|
|
|
f4e0315af9
|
Remove obsolete STAGE_D1.3.md and SMEM_P_GUIDANCE_REQUEST.md
|
2026-05-24 21:41:17 +00:00 |
|
|
|
dadfad8f89
|
Docs: Update STAGE_D.md, README.md with hd=512 compilation blocker, lessons learned
|
2026-05-24 21:35:25 +00:00 |
|
|
|
a5fef69363
|
D1.4: Use cutlass.range(unroll=1) for k_sub loops in both TMA and MMA warps
|
2026-05-24 17:55:33 +00:00 |
|
|
|
c11ac38ceb
|
D1.4: Remove --opt-level 0 from hd512 test (use default opt level)
|
2026-05-24 16:42:01 +00:00 |
|
|
|
b14d88f37f
|
D1.4: Fix merge test - use use_smem_p=False for hd=256 kernel (SMEM budget)
|
2026-05-24 16:36:48 +00:00 |
|
|
|
e6c9e6c0d0
|
D1.4: Add external k_sub merge test for hd=512 (avoids slow in-kernel k_sub compilation)
|
2026-05-24 16:31:06 +00:00 |
|
|
|
13fcf16b14
|
D1.4: Use --opt-level 0 only (ptxas -j not supported, MLIR is the bottleneck)
|
2026-05-24 15:43:17 +00:00 |
|
|
|
b4da412b30
|
D1.4: Use options string for compile flags (--ptxas-options -j64 --opt-level 0)
|
2026-05-24 15:40:39 +00:00 |
|
|
|
4f69dffc93
|
D1.4: Add PtxasOptions -j64 + OptLevel(0) for faster hd=512 compilation
|
2026-05-24 15:36:35 +00:00 |
|
|
|
331ddb29b7
|
D1.4: Fix regression test for un-normalized O output (D5a)
|
2026-05-24 15:13:16 +00:00 |
|
|
|
25201d0c3d
|
D1.4: Guard LSE computation with const_expr(not normalize) - fixes BF16 type mismatch in regression test
|
2026-05-24 15:11:39 +00:00 |
|
|
|
7f64a11eea
|
D1.4: Switch k_sub from cutlass.range to Python range (unrolled at trace time)
|
2026-05-24 15:10:28 +00:00 |
|
|
|
6d7b8fed3e
|
D1.4: Fix tTMrO placeholder - define only inside const_expr block
|
2026-05-24 14:23:22 +00:00 |
|
|
|
7a4ff959bf
|
D1.4: Use cutlass.range loop for k_sub (reduce IR), guard O rescale with const_expr(n_kv_tiles>1)
|
2026-05-24 14:22:45 +00:00 |
|
|
|
449a6e7ede
|
Fix: add cutlass import to test_d1_qk512
|
2026-05-24 14:20:32 +00:00 |
|
|
|
ce267909ad
|
Fix: add cpasync import to test_d1_qk512
|
2026-05-24 14:20:01 +00:00 |
|
|
|
625837fd44
|
D1.4: Add hd=512 QK-only and standalone test for compilation debugging
|
2026-05-24 14:19:26 +00:00 |
|
|
|
592873b560
|
D1.4: Reduce pv_n_tile to 128 for hd=512 to fit SMEM budget (192KB)
|
2026-05-24 08:07:32 +00:00 |
|
|
|
e7c146dbfd
|
D1: Unrolled k_sub path (hardcoded k_sub=0,1) to avoid cutlass.range IR explosion
|
2026-05-24 07:03:14 +00:00 |
|
|
|
dd39c2ebdf
|
D1: Use cutlass.range for k_sub loops (CuTeDSL immutable handle)
|
2026-05-24 06:43:30 +00:00 |
|
|
|
2bf3ee40aa
|
D1: Fix kvh scoping - define before loops, consume V via pipeline
|
2026-05-24 06:42:26 +00:00 |
|
|
|
f2170fc1b3
|
D1: Fix kvb→kvh typo in PV GEMM
|
2026-05-24 06:41:25 +00:00 |
|
|
|
e2b914be5e
|
D1: Remove qh.commit() - pipeline handles commit internally
|
2026-05-24 06:40:10 +00:00 |
|
|
|
583c509bcd
|
D1: TMA producer uses acquire_and_advance + commit (no wait_and_advance)
|
2026-05-24 06:38:15 +00:00 |
|
|
|
3bf1e62b58
|
D1: Use same pipeline API as working code (acquire_and_advance) for k_sub path
|
2026-05-24 06:36:19 +00:00 |
|
|
|
85af7f4cf3
|
D1: Add PipelineState for k_sub TMA path
|
2026-05-24 05:02:17 +00:00 |
|
|
|
622089ad16
|
D1: Fix pipeline API for K sub-tile path (producer_acquire/commit)
|
2026-05-24 04:59:41 +00:00 |
|
|
|
b9e806f09d
|
D1: K sub-tile MMA path using pipeline barriers
|
2026-05-24 04:57:08 +00:00 |
|
|
|
98e974403c
|
D1: Fix TMA copies in k_sub path (no mbarrier, use cp_async wait)
|
2026-05-24 04:53:46 +00:00 |
|
|
|
e637d3ae73
|
D1: Add K sub-tile loop for hd=512 (const_expr guarded, hd≤256 path unchanged)
|
2026-05-24 04:51:51 +00:00 |
|
|
|
24b9310682
|
D1: Debug TMA partition shapes at hd=512
|
2026-05-24 04:43:12 +00:00 |
|
|
|
9201a844dd
|
D1: K sub-tiling - qk_mma_tiler K-dim = k_tile=256, SMEM fits at hd=512
|
2026-05-24 04:41:12 +00:00 |
|
|
|
6be7690011
|
Docs: Update STAGE_D.md, README.md status for D1 hd≤256 milestone
|
2026-05-24 04:32:43 +00:00 |
|
|
|
787d0160a1
|
D1: Full test with TMEM-P at hd=64,128,256,512
|
2026-05-24 04:07:40 +00:00 |
|
|
|
d234297712
|
D1: Remove debug prints, clean up
|
2026-05-24 04:06:26 +00:00 |
|
|
|
3b63405ad4
|
D1: const_expr for sP layout selection (CuTeDSL)
|
2026-05-24 04:05:17 +00:00 |
|
|
|
1c8b043702
|
D1: Python if for sP layout (trace-time, not MLIR)
|
2026-05-24 04:04:27 +00:00 |
|
|
|
3aa8e5185a
|
D1: Tiny 4-mode sP placeholder for TMEM-P path
|
2026-05-24 04:03:28 +00:00 |
|
|
|
03ad730a9b
|
D1: Conditional sP allocation (saves 64KB SMEM for TMEM-P at hd=256)
|
2026-05-24 04:02:02 +00:00 |
|
