TileLang: высокоуровневый Python- DSL для проектирования производительных GPU-ядер
TileLang — Python-фреймворк на основе TVM, упрощающий написание эффективных GPU-ядер для GEMM, Softmax и FlashAttention с автоматической оптимизацией под архитектуру.
Открыть материал →