Skip to content

Writing study: TileLang: Bridge Programmability and Performance in Modern Neural Kernels

← ICLR 2026 oral writing corpus

Lei Wang, Yu Cheng, Yining Shi, Zhiwen Mo, Zhengju Tang, Wenhao Xie, Tong Wu, Lingxiao Ma, Yuqing Xia, Jilong Xue, Fan Yang, Zhi Yang · Official ICLR 2026 oral record

Figure, table and section ledger

Editorial contribution type: Inference, algorithms & systems efficiency (primary).

Layout extraction, not a required paper template. Counts are distinct caption labels; a multi-panel figure counts once. Section spans are estimated from heading positions, including floats and whitespace. Method and limitations.

Canonical PDF
36 physical pages
Before references / appendix
≈ 10.36 page-equivalents
Main figures / tables
10 / 0
Attached appendix figures / tables
9 / 19
Attached appendix span
≈ 21.54 page-equivalents
Before references / appendix: ≈ 28.78%References / intervening material: ≈ 11.39%Attached appendix: ≈ 59.83%Estimated layout envelopes, not text/artwork area. Unresolved appendix boundaries remain within the undivided post-body material.

Source-bound correction: TileLang: Figure 10 is a real side-wrapped caption on physical page 10. Poppler split Figure, 10:, and the caption body into separate same-baseline lines. Visually checked page 10; restore the caption rather than treating this as an author numbering gap.

Section-by-section lengths and source pages

The title/abstract/front matter precede the first numbered heading. A span ends at the next detected top-level heading, not at the last sentence; end-page whitespace is included. References and statements are listed separately. These are not measurements of text-only or artwork area.

Top-level headingStarts on PDF pageEstimated spanShare of entire PDF
1 INTRODUCTION1≈ 1.17 pp.3.25%
2 RELATED WORK2≈ 0.61 pp.1.69%
3 PROGRAMMING MODEL3≈ 1.18 pp.3.28%
4 SCHEDULING GUIDANCE AND AUTOMATION4≈ 3.27 pp.9.08%
5 EVALUATION7≈ 3.03 pp.8.42%
6 CONCLUSION10≈ 0.29 pp.0.81%
7 REPRODUCIBILITY STATEMENT11≈ 0.14 pp.0.39%
8 ACKNOWLEDGEMENT11≈ 0.12 pp.0.33%
REFERENCES11≈ 4.1 pp.11.39%
A EXTENDED DISCUSSION OF RELATED WORK15≈ 1.06 pp.2.94%
B MLA ALGORITHM16≈ 0.28 pp.0.78%
C PARTIAL LIST OF TILELANG PRIMITIVES16≈ 0.57 pp.1.58%
D PLATFORM-SPECIFIC SCHEDULING17≈ 1.11 pp.3.08%
E LAYOUT INFERENCE ALGORITHM18≈ 4.16 pp.11.56%
F PIPELINE INFERENCE ALGORITHM22≈ 2.53 pp.7.03%
G EFFECTIVENESS OF THE COST MODEL25≈ 0.42 pp.1.17%
H TUNING TIME25≈ 0.76 pp.2.11%
I MATMUL IMPLEMENTATION DIFFS: TVM VS. TILELANG VS. TRITON26≈ 0.38 pp.1.06%
J COMPARISON OF FTG IR AND TVM IR26≈ 1.38 pp.3.83%
K COMPARISON OF FLASHMLA IMPLEMENTATIONS ON DIFFERENT28≈ 0.61 pp.1.69%
L COMPARISON WITH RECENT SYSTEMS28≈ 1.71 pp.4.75%
M BASELINE COMMIT HASHES30≈ 0.31 pp.0.86%
N OPERATOR SHAPES IN OUR BENCHMARK30≈ 2.38 pp.6.61%
O KERNEL IMPLEMENTATIONS33≈ 3.89 pp.10.81%
All detected figure and table locations

Argument and organization

  1. Programming model

    A running FlashMLA example connects explicit programmer control to compiler automation.

  2. Scheduling and evaluation

    Recommendation/inference mechanisms precede cross-hardware performance, code-complexity comparisons, and ablations.

Problem, claim and contribution

The introduction presents performance and implementation complexity as a two-objective tension, then keeps both metrics visible throughout evaluation.

Pages 1–2, Figure 1. · Main paper

Argument and section order

A single MLA running example unifies programming primitives, graph representations, layouts, and pipeline scheduling instead of introducing unrelated examples.

Pages 4–7, sections 3.3–4.3. · Main paper

Figures, captions and visual explanation

Numbered callouts and red/blue dashed regions on executable-looking code distinguish recommended decisions from inferred details.

Page 4, Figure 2. · Main paper

Tables, comparisons and evidence

Small-multiple scatterplots repeat speedup versus lines-of-code axes across operators and separately group NVIDIA/AMD results; the caption tells readers which corner is desirable.

Page 9, Figure 8. · Main paper

Appendix and reproducibility

An appendix roadmap leads to pseudocode, hardware-specific code diffs, baseline commit hashes, benchmark shapes, and actual kernel implementations.

Page 15 roadmap; appendices E–O. · Main paper

Clarity and writing conventions

Explicit 'Goal and Scope' and 'Discussion' paragraphs translate long compiler algorithms into readable contracts.

Pages 18–25, appendices E–F. · Main paper

What to reuse—and what not to infer

Transferable lesson: Choose an executable running example and preserve its terminology across API explanation, intermediate representation, optimization mechanism, and empirical evaluation.

Boundary: Lines of code are a limited productivity proxy, and speedups depend on hardware, workload, baseline implementation, and tuning budget. Keep these qualifiers beside headline gains; dense code screenshots require larger zoomable versions.

Apply this lesson using the argument and evidence worksheet.

Inspection record and sources

Checked
2026-09-19
Reading status
Full text + selected visuals inspected
Main text
Pages 1–11, sections 1–8; references pages 11–14.
Appendix / supplement
Pages 15–36, appendices A–O: related work, primitive semantics, platform differences, algorithms, cost/tuning evaluation, baseline versions, shapes, and full kernel listings.
Visually inspected PDF pages
4, 9, 28