Show HN: Phobos – A tiny scale-free kernel language with tile-DAG support
5 days ago
- #GPU Programming
- #Machine Learning
- #Compiler Design
- 作者开发了Phobos,这是一种受Triton启发的微型内核语言,可编译为PTX并在NVIDIA GPU上运行。
- Phobos在RTX 2080 SUPER上实现了约76%的cuBLAS SGEMM性能,并采用了平铺和Tensor Core使用等优化。
- 该语言抽象了线程级细节,专注于面向AI计算的平铺级操作,从而能够自然地扩展到分布式集群。
- 该项目涉及学习底层GPU编程、使用MLIR和LLVM与Rust,并实现内核参数的自动调优。
- 为分布式平铺-DAG执行构建了集群原型,但由于硬件限制未进行多节点基准测试。