CROSS-LEVEL KERNEL PROGRAMMING

TileCasCascading Kernel Development
Across Programming Abstractions

Xinyu Xiao · Zihe Song · Dezhi Ran · Wei Yang · Tao Xie

Fudan University · The University of Texas at Dallas · Peking University
Beijing Tongming Lake Information Technology Application Innovation Center

Give coding agents the freedom to choose what to change,
where to change it, and which candidate to build on.

TileLang Native codeAscend & AMDAKO & KDA
HIGH LEVELTileLangStructure · Tiling · Data layout
TileCas
Diagnose · Schedule · RetainStay   /   Lower   /   Rollback
LOW LEVELAscend C / HIP C++Memory · Synchronization · Hardware

Lower exports a candidate. Rollback restores a retained High source.

THE PAPER

Abstractions for
agent-driven search.

Read the full abstract

Programming abstractions have traditionally been designed for humans. As coding agents take on more programming tasks, the best abstraction may depend on the stage of their search: high-level DSLs simplify exploration, while native code exposes finer control for performance tuning.

TileCas is a unified, pluggable interface for programming across abstraction levels. It combines Cross-Level Diagnosis, Abstraction Scheduling, and a Heterogeneous Search Tree to connect execution feedback with editable controls and retain useful search branches.

We integrate TileCas with AKO and KDA and evaluate both on 50 end-to-end KernelBench Level 3 workloads, across two accelerator platforms and two language models. The results show improvements across all four accelerator–model configurations.

4.84×

Mean speedup

AKO+TileCas over workload references

29.0%

Higher than the strongest baseline

Compared with AKO-Random by overall mean

69.2%

Improvement with KDA

The same interface, a second base workflow

01 / METHOD

One search, multiple programming levels.

Paper §3

The base workflow continues to generate, evaluate, and refine kernels. TileCas helps the agent choose the next modification, its source candidate, and its abstraction level.

TileCas workflow: task to planning, candidate generation, and evaluation; Cross-Level Diagnosis supports Abstraction Scheduling, with candidate history retained in a Heterogeneous Search Tree.
How TileCas connects to a base coding workflow. Diagnosis supplies evidence for a testable edit; scheduling chooses the source and level; the tree retains alternatives for future iterations. Original architecture: Figure 2 ↗
01

Cross-Level Diagnosis

Connect High-level calls to generated native code, then associate measurements with the candidate and executed kernel or call group.

What does the evidence support changing?

02

Abstraction Scheduling

Compare the controls available at each level and select a source candidate, action, and target level for a testable modification.

Where should the next edit happen?

03

Heterogeneous Search Tree

Retain candidates, derivation paths, and evaluation records across levels so useful alternatives can be revisited.

Which implementation should the agent resume?

EXPLORE THE ACTIONS

Three ways to continue.

Select an action to see how candidate history is used.

H₀→H₁→H₂

Refine within an existing representation

Edit a selected source at its current level. Stay can also resume a retained Low branch; it does not require continuing from the most recent candidate.

The agent makes these choices from evidence and history. The paper’s scheduling pseudocode describes agent decisions; it is not a separately trained routing model.

02 / EVALUATION

Explore the results.

Download Table 1 data

50 workloads · 4 accelerator–model configurations · 5 independent agent runs per pair.
Every run shares a 120-minute total search budget and the prescribed platform evaluator.

Performance across all four configurations

Table 1 · Mean endpoint speedup (×); overall SD is across five independent agent runs

Seed-level data ↓
GPT-6 Astra and DeepSeek-V4.1-Flash; overall equally weights the four configurations
MethodAscend / AstraAMD / AstraAscend / DSAMD / DSOverall ± SDCorrect
AKO-High2.4025.0131.0463.7923.063 ± 0.082188 / 200
AKO-Low3.0323.1531.9812.9152.770 ± 0.085192 / 200
AKO-Fixed2.9825.0552.2934.6323.741 ± 0.113188 / 200
AKO-Unrestricted2.4444.6961.8563.9423.235 ± 0.123192 / 200
AKO-Random3.7505.4871.9623.7923.748 ± 0.167191 / 200
KDA2.4293.0682.0072.9562.615 ± 0.044190 / 200
AKO+TileCas4.5766.7222.7785.2684.836 ± 0.221195 / 200
KDA+TileCas4.4376.3392.3714.5534.425 ± 0.119195 / 200

Correctness uses majority success (at least 3 of 5 runs). Mean endpoint speedups retain failed-run zeros. The interactive chart below provides each configuration’s uncertainty intervals.

End-to-end endpoint performance

Arithmetic-mean speedup over the matching workload reference

Loading results…

Overall weights the four configurations equally. Failures contribute zero.

View numerical results and uncertainty
Overall performance
MethodMean ± SDMedian95% bootstrap CIMajority correct
SUPPLEMENTARY EVIDENCE

Do the gains survive final verification?

Measured re-evaluation of the selected implementations · same aggregation and reference convention

Comparison data ↓

Yes. Using final-verification results for every method preserves the principal overall mean-performance comparisons. Best-reported search endpoints remain the paper’s reporting convention; the measurements below provide a separate sensitivity analysis.

<0.56%Absolute change in each method’s overall mean speedup
0.65–1.05%Range of per-method median absolute latency changes
>90%Successful runs within ±3% latency change in every method
Relative gains in overall arithmetic-mean speedup; paired outcomes use the same 3% tie rule
ComparisonBest-reported gainFinal-verify gainBest W / T / LFinal W / T / L
AKO+TileCas vs AKO-Random29.04%29.42%158 / 5 / 37152 / 9 / 39
AKO+TileCas vs AKO-Fixed29.29%29.39%144 / 6 / 50135 / 17 / 48
KDA+TileCas vs KDA69.22%69.51%159 / 3 / 38159 / 4 / 37

Individual paired wins, ties, and losses can change even though the principal mean-performance conclusions remain stable. Across all 55 method–configuration summaries, the largest absolute mean-speedup change is 0.679%. Existing confidence intervals describe the original search endpoints only.

View best-reported and final-verification means for all methods
Overall mean speedup and relative change; all 11 methods and variants
MethodBest reportedFinal verificationMean changeRuns within ±3% latency
AKO+TileCas4.836×4.823×-0.266%90.83%
AKO-Fixed3.740×3.728×-0.342%95.40%
AKO-High3.063×3.058×-0.170%95.18%
AKO-Low2.770×2.768×-0.090%95.47%
AKO-Unrestricted3.234×3.219×-0.474%94.64%
KDA2.615×2.605×-0.371%91.28%
KDA+TileCas4.425×4.416×-0.198%95.14%
AKO-Random3.748×3.727×-0.555%93.47%
w/o Diagnosis4.229×4.215×-0.316%95.21%
w/o Scheduling3.854×3.845×-0.245%96.06%
w/o Tree4.141×4.130×-0.274%93.41%

How the search progresses

Mean best-so-far speedup at each reporting budget · Figure 6

Post-step curves use the supplied reporting coordinates. A run scores zero before first correctness and carries its last value forward after termination. Download coordinates ↓

Stronger endpoints, not earlier first correctness.

High reaches the first correct implementation in 6.81 minutes on average, compared with 10.74 minutes for TileCas. TileCas’s advantage appears during continued refinement.

Each component contributes.

Removing Diagnosis, Scheduling, or the Search Tree reduces mean endpoint speedup by 12.6%, 20.3%, and 14.4%, respectively. These ablation losses are not additive.

Browse results for individual workloads 2,200 method–workload–configuration pairs
Best-reported and final-verification means over five agent runs; correctness count is shown separately
WorkloadBest reportedFinal verificationCorrect runsMajority success
03 / CASE STUDIES

Follow the search, candidate by candidate.

Figure conventions
ASCEND / ASTRA Paper §4.3

MLP: recovering from an early failure

TileCas’s first High-level candidate fails compilation. Iteration 2 repairs it before any Lower or Rollback action.

High-stage baseline4.13 minFirst correctness
TileCas5.41 minFirst correctness

The High-stage trace is the High prefix of the Fixed experiment. Its final case speedup is 1.815×; TileCas’s is 2.123×. These are single-case results, separate from Table 1 aggregates.

AMD / ASTRA Paper §3 & §4.5

MinGPT: returning to a useful branch

Retained High candidates reopen structural choices. Later Low candidates can reuse compatible native code from earlier branches.

Selected search best3.831 msIteration 53
Final verification4.138 msRestored selected source

Unchanged repeat, iter-58: 3.880 ms. The same source and configuration passed correctness again, with a 1.28% latency difference from iter-53. The later 4.138 ms verification also passed. Read the measurement note ↓

Figure 4 uses the case’s own 44.954 ms reference (11.734× best; 10.864× verification). Figure 5 uses the same 42.969 ms reference as its three baselines (11.215× best). The approximately 8% candidate remeasurement difference is independent of this normalization. Read the clarification ↗

42 / 50

Selected cross-level actions outperform Stay at sampled checkpoints.

Matched interventions start from the same candidate state with the same 10-evaluation local budget: 23/25 High checkpoints favor Lower; 19/25 Low checkpoints favor Rollback. This is evidence about the sampled switching states and tested alternatives.

04 / READING GUIDE

The details behind
the headline results.

Supplementary explanations of the method, measurement contract, and scope of the evidence.

Download this guide
Why do Figures 4 and 5 use different reference latencies?

Figure 4 reports the MinGPT case using its own reference latency of 44.954 ms. Figure 5 uses a common reference of 42.969 ms shared with its three baseline trajectories, so those curves are compared on the same normalization basis. The reference convention is therefore not fully uniform across the two figures: their speedup numbers should be interpreted with their respective denominators.

The selected iter-53 source took 3.831 ms during search, 3.880 ms in an unchanged repeat at iter-58, and 4.138 ms in final verification. The retained note records identical source and configuration and successful correctness checks in all three evaluations. The repeat differs by 1.28%; the approximately 8% best-to-final increase is a candidate remeasurement difference, independent of reference normalization. It is consistent with between-evaluation variability and does not reflect a source change during restoration.

Final verification still gives a 10.864× case speedup. The new aggregate sensitivity analysis also preserves the principal overall mean-performance conclusions when every method uses final-verification values. The Figure 4 case is recorded separately: its mapping to a main-cohort run has not been established, so it is not included in that aggregate. Original figures and numerical records are unchanged. Case alignment ↓ · Figure conventions ↗

What exactly is a reported endpoint?

The main tables use best reported: the fixed matching reference latency divided by the lowest eligible median latency from a correct formal search evaluation. Each median uses five timed calls after three warmups. Compile failures, incorrect candidates, diagnostic-only measurements, and protocol-invalid evaluations are excluded; a run with no eligible correct candidate scores zero.

Final verification re-evaluates the restored selected source and is recorded separately. Its latency does not replace the search endpoint. A best-reported speedup therefore describes the search record and need not equal the speedup measured in a later verification. Endpoint definition ↗

What do “five runs,” correctness, SD, and CI mean?

Five independent agent runs are separate from the five timed forward calls within one evaluation. Correctness uses shared standard-normal floating inputs, ten trials for each seed base 42, 43, and 50000, and atol = rtol = 0.001. Timing uses input seed 42. Agent seed labels are 101, 211, 307, 401, and 503.

A workload–configuration pair meets majority correctness if at least three of five runs succeed. Endpoint scores are averaged over all five runs, including failed-run zeros; a pair with fewer than three successes can still contribute a positive endpoint mean.

The overall mean equally weights the four configurations. SD is across the five aggregate repeat means (ddof = 1). The 95% confidence intervals use 20,000 paired workload bootstrap draws, retaining each workload’s configurations and repeats together. They describe uncertainty conditional on the supplied endpoint records. Resampling settings ↗

Are AMD and Ascend measuring the same latency boundary?

Both use the same platform-specific boundary for reference and candidate. AMD uses a synchronized HIP-event interval around the forward invocation. Ascend uses the msprof device-task envelope, from the first attributed task start to the last attributed task completion.

Both include intervening gaps and count overlap once. Ascend excludes host work outside that device interval. These boundaries differ, so normalized speedups should not be read as a direct comparison of absolute hardware latency or identical host request latency. Per-task duration sums are diagnostic only. Timing protocol ↗

Does Rollback translate edited native code back to TileLang?

No. Rollback restores a retained higher-level source and continues editing there, while preserving the existing Low branch. Lower exports a supported native representation from a selected High candidate. Stay edits a selected source in its existing representation and can resume a retained Low branch.

The tree gives each snapshot one derivation parent. Cross-branch code reuse is recorded in iteration history and source records. A native candidate’s High parent records its origin; after native edits, that lineage alone does not establish an exact source correspondence.

What does diagnosis establish, and who makes the decisions?

Tools connect calls, generated source, argument bindings, and available measurements for a particular candidate. The coding agent interprets those observations, judges whether evidence supports a concrete modification, and chooses the edit, candidate, and level. The functions in Algorithm 1 describe that agent process; explicit enumeration or numerical ranking of all plans is not required.

A kernel or call-group timing measures that whole region. For example, Figure 3’s 2.786 ms is the whole attention call, not the reduction or barrier alone. Code inspection can motivate an optimization hypothesis, but attributing a measured cost to an individual operation requires additional evidence. Ambiguous measurement-to-code matches remain unassociated.

Was the High-only search stopped early to start the Low stage?

The High-only series in Figures 1 and 5 is the complete High-stage AKO invocation also used as Fixed’s first stage. It terminated autonomously under AKO’s own stopping rules at approximately 14.72 minutes; this was its observed completion time, not an experimenter-imposed cutoff. During that invocation, the agent worked exclusively in TileLang and was not instructed to reserve work for Low.

Only after High completed did the controller export its selected implementation and start a fresh Low-stage session without the High conversation. The High-only and High-to-Low curves therefore share one completed High run rather than independent High samples. Under the same High-stage configuration and stopping rules, that invocation is operationally equivalent to standalone High-only AKO.

The 1.264× endpoint characterizes this search trajectory, not a limit of TileLang’s expressiveness. With different history and guidance, TileCas’s High iter-17 reaches approximately 8.345× under the common reporting reference. Autonomous stopping establishes completion under a workflow policy, not exhaustion of all useful High-level optimizations. High-only search and autonomous termination ↓

What do Fixed, Random, and the ablations control for?

Fixed runs independent High and Low agent conversations, separated by a validated export; the Low agent inherits neither the High conversation nor its history. Unrestricted retains AKO’s existing representation choices under the platform contract.

Random samples legal cross-level actions using transition frequencies estimated from this evaluation cohort. After a sampled Rollback, the agent still chooses a retained correct High source. It is a post-hoc frequency control, not a fully random candidate selector or an out-of-sample routing policy.

w/o Scheduling replaces scheduling with a fixed High-to-Low progression inside the component ablation; it should not be assumed identical in all other respects to the independent-agent Fixed baseline. Ablation effects interact and should not be added together.

How strong is the evidence that Rollback helps?

In the next three evaluations, 64.4% of eligible Rollback windows improve the incumbent by more than 3%. Extending the window until the next Rollback or termination raises the productive fraction to 79.8%, with a 50.07% median gain. This extended result includes later edits and is a descriptive association, not an isolated causal effect of the Rollback itself.

The matched interventions provide a more controlled comparison: from the same state and with the same 10-evaluation budget, Rollback beats Stay at 19/25 sampled Low checkpoints. These results support the tested choices at those checkpoints, not a universal guarantee that Rollback is always preferable.

What is available, and what remains outside the evaluation?

The release includes workflow and backend source, evaluator support, 11,000 endpoint records across 11 methods or variants, final-verification values, timing samples, resampling settings, candidate snapshots, and case traces. KDA’s upstream resources remain external; the package includes local adapters and setup instructions. Its dependency metadata distinguishes the supplied reproduction configuration from historical experiment provenance. KDA integration details ↗

The named MLP and MinGPT cases include candidate and output archives. The 500-episode Rollback table supplies event ordering, but lacks original run IDs and per-episode candidate-source or raw-log paths. Likewise, main-cohort publication run IDs are not mapped to historical run directories. These materials support different levels of inspection; the event table alone is not a complete source archive for all 500 episodes. Coverage of retained records ↗

The evidence covers two editable levels, 50 KernelBench Level 3 workloads, two accelerator platforms, and two models. Generalization to additional levels, dynamic shapes, communication-heavy workloads, multi-device execution, or future compiler versions remains to be evaluated.

05 / OPEN MATERIALS

From the paper to the evidence.

Download all materials

The six bundles cover the paper’s data-availability references. Each includes its original documentation and supporting records. Current development is available on GitHub. View the paper-to-file index ↗

View platform and software versions
Source-package inspection dated 5 October 2026; historical case records are included in the software bundle
PlatformHigh representationNative toolchain
Huawei Ascend 910B3TileLang-Ascend 0.1.4 (imported source)Ascend C / CANN 9.0.0
AMD Radeon AI PRO R9700TileLang 0.1.14ROCm SDK 7.14.1; HIP 7.14.60850

An installed Ascend wheel reports 0.1.4+ubuntu.20.4.cann900; its metadata is distinct from the imported source module. The system-default CANN version is not the case’s selected toolchain. Read the version evidence ↗

REFERENCE

Cite this manuscript

Use the citation below for the manuscript and project.

@misc{xiao2026tilecas,
  title = {TileCas: Cascading Kernel Development Across Programming Abstractions},
  author = {Xinyu Xiao and Zihe Song and Dezhi Ran and Wei Yang and Tao Xie},
  year = {2026},
  eprint = {2610.14740},
  archivePrefix = {arXiv},
  primaryClass = {cs.SE},
  url = {https://arxiv.org/abs/2610.14740}
}