Eä: 러스트로 작성된 파이썬용 SIMD 컴파일러

Eä의 기능

Eä는 SIMD 커널용 컴파일러로, 일반적인 워크플로 문제를 해결합니다: Python 코드 작성, 핫 루프 찾기 위한 프로파일링, C로 재작성, ctypes와 씨름, 포인터 디버깅, 결국 5배 속도 향상 달성—그리고 다음 주에 이 과정을 반복하는 문제 말이죠.

작동 방식

Eä를 사용하면 다음과 같습니다:

• 작은 .ea 파일 작성
• 한 명령 실행
• 일반 함수처럼 Python에서 호출

커널은 네이티브 벡터화 속도로 실행됩니다. 사용 예시:

import ea
kernel = ea.load("fma.ea")
result = kernel.fma_f32x8(a, b, c, out)  # NumPy보다 6.6배 빠름

기술적 세부사항

컴파일러는 다음을 생성합니다:

• 공유 라이브러리
• Python 래퍼
• Rust, C++, PyTorch, CMake 바인딩도 포함

대상 아키텍처:

• x86-64 (AVX2 / AVX-512)
• AArch64 (NEON)

컴파일러 자체:

• 약 12,000줄의 Rust 코드
• 475개 테스트
• ctypes, 헤더 파일, 빌드 시스템 불필요

개발 배경

개발자는 AI 모델의 도움을 받아 Eä를 구축했으며, 아키텍처 제어를 유지하면서 Claude를 주요 작업에 활용했습니다. 주요 통찰은 SIMD 자체보다 모든 "접착 코드" 처리가 더 중요하다는 점이었으며, 이로써 개발자는 커널에만 집중할 수 있습니다.

벤치마크

벤치마크는 fma_f32x8 예제에서 NumPy보다 6.6배 빠른 성능을 보여줍니다. 개발자는 이 결과가 상당히 단순한 설정에서 나왔지만, 공정하고 재현 가능하도록 유지하려 했다고 언급했습니다.