English

Demonstrating a Future for MLIR-native DSL Compilers on a NumPy-like Example

Programming Languages 2026-04-28 v2

Abstract

Compilers for general-purpose languages have been shown to be at a disadvantage when it comes to specialized application domains as opposed to their Domain-Specific Language (DSL) counterparts. However, the field of DSL compilers features little consolidation in terms of compiler frameworks and adjacent software ecosystems. As a result, considerable work is duplicated, lost to maintenance issues, or remains undiscovered, and most DSLs are never considered "production-ready". One notable development is the introduction of the Multi-Level Intermediate Representation (MLIR), which promises a similar impact on DSL compilers as LLVM had on general-purpose tooling. In this work, we present a NumPy-like DSL made for offloading numeric tensor kernels that is entirely MLIR-native. In a first for open-source, it implements all frontend actions and semantic analyses directly within MLIR. Most notably, this is made possible by our new dialect-agnostic MLIR type checker, created for the future of DSLs in MLIR. We implement a simple, yet effective, parallel-first lowering scheme that connects our language to another MLIR dataflow dialect for seamless offloading. We show that our approach performs well in real-world use cases from the domain of weather modeling and Computational Fluid Dynamics (CFD) in Fortran.

Keywords

Cite

@article{arxiv.2604.19906,
  title  = {Demonstrating a Future for MLIR-native DSL Compilers on a NumPy-like Example},
  author = {Karl F. A. Friebel and Jascha A. Ohlmann and Jeronimo Castrillon},
  journal= {arXiv preprint arXiv:2604.19906},
  year   = {2026}
}

Comments

17 pages, 5 figures

R2 v1 2026-07-01T12:29:12.732Z