From e632490682032b56d96d48d308a033ab6ff960f6 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sun, 24 May 2026 00:37:36 +0000
Subject: [PATCH] D1.5: Add TODO for correction epilog - keeping working TMEM
 round-trip for now

---
 dsv4/kernels/attention/fmha.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dsv4/kernels/attention/fmha.py b/dsv4/kernels/attention/fmha.py
index 46ba5611..d3ed7cbe 100644
--- a/dsv4/kernels/attention/fmha.py
+++ b/dsv4/kernels/attention/fmha.py
@@ -410,6 +410,7 @@ class FmhaKernel:
             final_o_bar.arrive_and_wait()
 
             # === NO-OP TMEM round-trip: re-map O from MMA layout to epilog layout ===
+            # TODO: Replace with correction epilog (D1.5) for zero-error one-way trip
             tTMrO_noop = cute.make_rmem_tensor(
                 (tTMEM_LOADcO.shape, 128 // corr_tile_size), self.acc_dtype
             )