Backend
CompileMask: Bridging the Gap Between Expressiveness and Efficiency in Attention Masking
arXiv cs.DC0 views
arXiv:2610.13474v1 Announce Type: new Abstract: Modern attention workloads require diverse masking patterns whose boundaries depend on runtime metadata, yet efficient attention kernels natively support only a small set of patterns. Existing methods for supporting custom masks in these kernels face an expressiveness-efficiency trade-off. Predicate-based methods are expressive but incur substantial preprocessing and execution overhead, whereas interval-based methods execute efficiently but restrict expressiveness through a predefined interval count. We present CompileMask, which addresses this trade-off by connecting expressive mask specifications with efficient attention execution. We design a compiler that analyzes visibility rules during the offline stage and generates a mask-specific preprocess kernel. We further propose MaskSpec, a region-based DSL that lets users express these rules. During the online stage, the generated kernel reads batch metadata and produces a compact Execution Descriptor that lets the attention kernel efficiently check visibility by comparing element indices against precomputed endpoints. Experiments show that CompileMask achieves up to $3.8\times$ speedup over state-of-the-art methods.
Source:https://arxiv.org/abs/2610.13474PreviousShoreKeeper: Accelerating Iterative Data Analytics in the Cloud via Distributed Recovery-Cost-Aware CachingNextGPU-resident Execution for Timing-predictable Sensor Processing Systems
Related posts

The latest AI news we announced in October 2026
0Google AI Blog

Google Cloud Spend Caps Reach General Availability With an API and Live Costs
0dev.to
I built a Tinder-style Website for finding coding project ideas
0dev.to
GSAP ScrollTrigger image sequence in WordPress without code: when DIY isn't worth it
0dev.to
Surviving AWS Spot Interruptions With Real Checkpointing
0dev.to
Forward Watch
0dev.to