kernels/rr: staged-image Swin layer engine prototypes
Per-layer staged-image engine for the eleven translated denoiser Swin layers: descriptors (verified offsets per phase), cut recipes that splice an isolated native phase into NVIDIA's PTX, wave templates, epilogue notes and GPU probes. The isolated phase-8 splice produces a byte-identical final 4K frame but is slower than the translated body (14.90 vs 6.59 ms) - a correctness experiment, not a shipping improvement; docs/rr-engine.md records the design and the next direct-converter win (B operands built from transposed accumulators).
This commit is contained in:
1 parent
798e25611c
commit
0fe3a28cb7
58 files changed
+28330
-5
No files matched your search
@@ -784,7 +784,7 @@ def print_cut_abi(stmts, phases, defs, entry, path, lines, body):
|
||||
live = live_across(stmts, lo, hi)
|
||||
print(f" returns {len(live)} accumulators written in the region and "
|
||||
f"read after s{hi}")
|
||||
print(f" {live[:12]}{' ...' if len(live) > 12 else ''}")
|
||||
print(f" {live}")
|
||||
weight_b = sorted({m.B for p in phases for m in p.mma})
|
||||
print(f" weight image {len(weight_b)} distinct B fragments; byte ranges "
|
||||
f"per phase are in the table above")
|
||||
|
||||
Reference in new issue
Block a user