Repolode
triton-lang

triton-lang/triton

Python과 비슷한 코드로 CUDA보다 생산성 높게 고효율 딥러닝 GPU 커널을 작성할 수 있는 언어이자 컴파일러입니다.

대체할 수 있는 제품:CUDA

스타
20,529
포크
3,376
열린 이슈
1,372
마지막 커밋
10시간 전

어떤 프로젝트인가요?

Triton은 GPU에서 실행되는 저수준 커널인 커스텀 딥러닝 프리미티브를 작성하기 위한 언어이자 컴파일러입니다. CUDA보다 생산성 높게 빠른 커널을 작성하면서 다른 도메인 특화 언어보다 유연성을 높이는 것을 목표로 합니다.

이 프로젝트는 pip에서 CPython 3.10~3.14용 바이너리 휠로 설치할 수 있으며, 커스텀 또는 사전 빌드된 LLVM을 사용해 소스에서 빌드할 수도 있습니다. NVIDIA GPU(Compute Capability 8.0 이상)와 AMD GPU(ROCm 6.2 이상)를 대상으로 하며, CPU 지원은 개발 중인 것으로 안내되어 있습니다. 지원 플랫폼은 Linux입니다.

저장소에는 GPU 없이 커널을 실행하고 커널 코드 안에서 Python 중단점을 사용할 수 있는 Triton 인터프리터도 포함되어 있습니다. 환경 변수 설정을 통해 각 MLIR 및 LLVM 패스의 IR 덤프, 커널 재정의, 재현기 생성, 디버깅과 성능 분석을 위한 자동 튜닝 결과를 확인할 수 있습니다.

핵심 기능

  • 01타일 수준 추상화를 제공하는 Python 기반 언어로 GPU 커널을 작성합니다.
  • 02NVIDIA 및 AMD GPU에서 커널을 실행합니다.
  • 03GPU 없이 실행하고 Python 중단점을 지원하는 인터프리터 모드를 포함합니다.
  • 04검토를 위해 MLIR 및 LLVM 컴파일 단계마다 IR을 덤프합니다.
  • 05커널 재정의와 트리 외부 컴파일러 플러그인을 지원합니다.

활용 사례

  • 딥러닝 모델용 커스텀 융합 연산자나 attention 커널을 작성합니다.
  • 직접 CUDA로 작성하는 것보다 생산성을 높여 GPU 커널을 최적화합니다.
  • GPU가 없는 머신에서 인터프리터로 커널 로직을 디버깅합니다.
  • GPU 코드 생성을 위한 MLIR 기반 컴파일러 파이프라인을 살펴보거나 확장합니다.

빠른 시작

  1. 1

    pip에서 최신 안정 버전을 설치합니다

    pip install triton
  2. 2

    또는 소스에서 설치합니다: 저장소를 클론합니다

    git clone https://github.com/triton-lang/triton.git
  3. 3

    소스 트리에서 빌드 시 필요한 종속성을 설치합니다

    pip install -r python/requirements.txt
  4. 4

    Triton을 editable mode로 설치합니다

    pip install -e .

사용 전 알아둘 점

  • 소스에서 빌드하려면 로컬 LLVM 툴체인을 설정하거나 빌드 중 사전 빌드된 LLVM을 다운로드해야 합니다.
  • GPU 테스트 스위트를 실행하려면 GPU가 필요하지만, 인터프리터 모드는 GPU 없이 실행할 수 있습니다.

포함된 컬렉션

토픽

자주 묻는 질문

triton은(는) 무엇인가요?
Python과 비슷한 코드로 CUDA보다 생산성 높게 고효율 딥러닝 GPU 커널을 작성할 수 있는 언어이자 컴파일러입니다. Triton은 GPU에서 실행되는 저수준 커널인 커스텀 딥러닝 프리미티브를 작성하기 위한 언어이자 컴파일러입니다. CUDA보다 생산성 높게 빠른 커널을 작성하면서 다른 도메인 특화 언어보다 유연성을 높이는 것을 목표로 합니다.
triton은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
triton은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
triton은(는) 아직 유지보수되고 있나요?
활발히 유지보수 중. triton의 마지막 커밋은 2026년 11월 6일입니다.
triton은(는) 어디에 사용할 수 있나요?
딥러닝 모델용 커스텀 융합 연산자나 attention 커널을 작성합니다. 직접 CUDA로 작성하는 것보다 생산성을 높여 GPU 커널을 최적화합니다. GPU가 없는 머신에서 인터프리터로 커널 로직을 디버깅합니다. GPU 코드 생성을 위한 MLIR 기반 컴파일러 파이프라인을 살펴보거나 확장합니다.
triton는 어떻게 설치하나요?
pip에서 최신 안정 버전을 설치합니다: pip install triton 또는 소스에서 설치합니다: 저장소를 클론합니다: git clone https://github.com/triton-lang/triton.git 소스 트리에서 빌드 시 필요한 종속성을 설치합니다: pip install -r python/requirements.txt Triton을 editable mode로 설치합니다: pip install -e .
triton은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/triton-lang/triton

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트

LLVM을 사용해 수치 연산용 Python 함수를 빠른 머신 코드로 변환하는 NumPy 지원 오픈소스 JIT 컴파일러입니다.

1.1만+3 이번 주1.4천PythonBSD-2-Clause활발히 관리됨업데이트 어제

HGEMM, FP8/FP4 GEMM, Flash-Attention을 다루는 200개 이상의 커널이 포함된 최신 CUDA 학습 노트 오픈소스 책 및 코드 모음입니다.

학습 자료#cuda#gpu#tensor-cores
1.2만+17 이번 주1.3천CudaGPL-3.0업데이트 그저께