Use triton wheel no fork#2959

Merged

Dewei-Wang-sh merged 19 commits into

mainfrom

use_triton_wheel_no_fork

Apr 30, 2026

Commits on Apr 29, 2026

CI: try downloading pre-built Triton wheel before building from source

mengfei-jiang
and
claude
committed
CI: add rocm version to pre-built Triton wheel filename

mengfei-jiang
and
claude
committed
CI: URL-encode '+' as '%2B' in Triton wheel download URL

mengfei-jiang
and
claude
committed
CI: fix Triton wheel download URL path

mengfei-jiang
and
claude
committed
Use requirements-triton.txt to install triton from AMD PyPI

mengfei-jiang
and
claude
committed
add torch compile test for triton
mengfei-jiang
committed
Fix black formatting in torch_compile tests

mengfei-jiang
and
claude
committed
Fix fused_mul_add bf16 tolerance and add fullgraph=True

mengfei-jiang
and
claude
committed
Fix black formatting for fullgraph=True line wrap

mengfei-jiang
and
claude
committed
Use fp16/bf16 dtypes, add dynamo.reset, relax bf16 tolerance

mengfei-jiang
and
claude
committed
Replace manual triton install with amd-triton from PyPI

mengfei-jiang
and
claude
committed
Remove PIP_EXTRA_INDEX_URL for AMD PyPI in CI script

mengfei-jiang
and
claude
committed
Wrap amd-triton uninstall in try/except to avoid setup failure

mengfei-jiang
and
claude
committed
Revert to requirements-triton.txt approach with amd-triton from PyPI

mengfei-jiang
and
claude
committed
Auto-detect ROCm version and install amd-triton from AMD PyPI

mengfei-jiang
and
claude
committed
Add amd-triton install to CI workflows and fix PEP 517 build failure

mengfei-jiang
and
claude
committed
Extract triton install logic into shared install_triton.sh script

mengfei-jiang
and
claude
committed
Use install_triton.sh in setup.py instead of inline logic

mengfei-jiang
and
claude
committed
Deduplicate _get_compiled helper, add torch_compile FILE_TIMES, and install amd-triton in aiter-test CI

mengfei-jiang
and
claude
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Use triton wheel no fork#2959

Use triton wheel no fork#2959
Dewei-Wang-sh merged 19 commits into
mainfrom
use_triton_wheel_no_fork

Commits on Apr 29, 2026

CI: try downloading pre-built Triton wheel before building from source

CI: add rocm version to pre-built Triton wheel filename

CI: URL-encode '+' as '%2B' in Triton wheel download URL

CI: fix Triton wheel download URL path

Use requirements-triton.txt to install triton from AMD PyPI

add torch compile test for triton

Fix black formatting in torch_compile tests

Fix fused_mul_add bf16 tolerance and add fullgraph=True

Fix black formatting for fullgraph=True line wrap

Use fp16/bf16 dtypes, add dynamo.reset, relax bf16 tolerance

Replace manual triton install with amd-triton from PyPI

Remove PIP_EXTRA_INDEX_URL for AMD PyPI in CI script

Wrap amd-triton uninstall in try/except to avoid setup failure

Revert to requirements-triton.txt approach with amd-triton from PyPI

Auto-detect ROCm version and install amd-triton from AMD PyPI

Add amd-triton install to CI workflows and fix PEP 517 build failure

Extract triton install logic into shared install_triton.sh script

Use install_triton.sh in setup.py instead of inline logic

Deduplicate _get_compiled helper, add torch_compile FILE_TIMES, and install amd-triton in aiter-test CI