CODA: Rewriting Transformer Blocks as GEMM-Epilogue Programs
- It maintains local state for operations such as the online log-sum-exp used in cross-entropy loss.
- It recovers a substantial portion of GPU time that would otherwise be lost to data movement.
- It provides a structured DSL that simplifies writing high-performance kernels.