Hasty Briefsbeta

双语

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执行构建了集群原型,但由于硬件限制未进行多节点基准测试。