Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Mirage: A Multi-Level Superoptimizer for Tensor Programs

Created by
  • Haebom

저자

Mengdi Wu, Xinhao Cheng, Shengyu Liu, Chunan Shi, Jianan Ji, Kit Ao, Praveen Velliengiri, Xupeng Miao, Oded Padon, Zhihao Jia

개요

Mirage는 텐서 프로그램을 위한 최초의 다단계 슈퍼 최적화기입니다. GPU 컴퓨팅 계층 구조의 커널, 스레드 블록, 스레드 레벨에서 텐서 프로그램을 통합적으로 표현하는 μGraphs를 사용합니다. μGraphs를 통해 Mirage는 대수적 변환, 스케줄 변환 및 새로운 사용자 정의 커널 생성을 결합한 새로운 최적화를 발견할 수 있습니다. 방대한 검색 공간을 탐색하기 위해 추상화 기반 가지치기 기법을 도입하여 검색 공간을 크게 줄이고 특정 최적성을 보장합니다. 최적화된 μGraph가 입력 프로그램과 동등함을 보장하기 위해 강력한 이론적 보장을 갖춘 확률적 동등성 검증 절차를 도입합니다. 평가 결과, Mirage는 널리 사용되고 최적화된 DNN에 대해서도 기존 방법보다 최대 3.3배의 성능 향상을 보였습니다. Mirage는 https://github.com/mirage-project/mirage 에서 공개적으로 이용 가능합니다.

시사점, 한계점

시사점:
GPU 텐서 프로그램에 대한 최초의 다단계 슈퍼 최적화기를 제시.
μGraphs를 통해 대수적 변환, 스케줄 변환, 커널 생성을 통합적으로 최적화 가능.
추상화 기반 가지치기 기법으로 효율적인 검색 공간 탐색.
확률적 동등성 검증 절차를 통해 최적화 결과의 정확성 보장.
기존 최적화 방법 대비 최대 3.3배의 성능 향상.
오픈소스로 공개되어 접근성 향상.
한계점:
확률적 동등성 검증 절차의 신뢰도에 대한 추가적인 연구 필요.
다양한 유형의 텐서 프로그램에 대한 일반화 가능성에 대한 추가적인 검증 필요.
가지치기 기법의 최적화된 매개변수 설정에 대한 연구 필요.
👍