tilelang, 고성능 커널 개발용 언어 공개
tile-ai의 tilelang은 GPU, CPU 및 가속기 커널 개발을 간소화하는 도메인 특화 언어입니다. 이 언어는 고성능 컴퓨팅 분야에서의 개발 효율성을 높이기 위해 설계되었습니다.
- ·tilelang은 고성능 컴퓨팅 커널 개발을 간소화합니다.
- ·화웨이 Ascend 950 NPU를 공식 지원합니다.
- ·멀티 백엔드 언어 방식으로 다양한 하드웨어 경로 지원.
tile-ai가 개발한 tilelang은 GPU, CPU 및 가속기 커널 개발을 간소화하기 위한 도메인 특화 언어입니다. 이 언어는 고성능 컴퓨팅 분야에서 개발자들이 효율적으로 작업할 수 있도록 설계되었습니다. tilelang은 Python과 유사한 문법을 사용하며, TVM 위에 구축된 컴파일러 인프라를 활용하여 최신 성능을 위한 저수준 최적화를 가능하게 합니다.
최근 tilelang은 화웨이 Ascend 950 NPU를 공식 지원하게 되었습니다. 이를 통해 네이티브 코드 생성, 자동 스케줄링 및 동기화, SIMD/SIMT 벡터 프로그래밍 등을 지원하며, Ascend 예제를 통해 GEMM, FlashAttention 등의 기능을 탐색할 수 있습니다. 또한, tilelang은 다양한 하드웨어 경로를 지원하는 멀티 백엔드 언어 방식을 도입하여 개발자들이 더 많은 선택지를 가질 수 있도록 하고 있습니다.
tilelang은 또한 언어 서버 프로토콜(LSP) 구현을 오픈 소스로 공개하여 버퍼 모양, 데이터 타입, 범위 및 추론된 레이아웃에 대한 인레이 힌트를 제공하고, 정확한 진단 기능을 강화했습니다. 이 외에도 CUDA와 Metal 하드웨어 경로를 추가하고, 여러 레거시 API를 제거하여 호환성 노트를 참고하여 업그레이드할 수 있도록 했습니다.
관련 뉴스
관련 소식 찾는 중…